arupchakraborty2004 commited on
Commit
c5c7526
·
verified ·
1 Parent(s): d2a93d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,6 +32,8 @@ def denoise_image(noisy_image, strength: float = 0.7):
32
  return result.images[0]
33
 
34
  def inpaint_image(base_image, mask_image):
 
 
35
  base_image = base_image.convert("RGB").resize((512, 512))
36
  mask_image = mask_image.convert("RGB").resize((512, 512))
37
  prompt = "A completed and visually realistic image with the missing parts filled in."
@@ -51,8 +53,6 @@ def process_image(task, image, mask=None):
51
  elif task == "Denoise":
52
  return denoise_image(image)
53
  elif task == "Inpaint":
54
- if mask is None:
55
- raise ValueError("Please upload a mask for inpainting.")
56
  return inpaint_image(image, mask)
57
  elif task == "Upscale":
58
  return upscale_image(image)
@@ -69,7 +69,7 @@ interface = gr.Interface(
69
  value="Colorize"
70
  ),
71
  gr.Image(type="pil", label="Upload Image"),
72
- gr.Image(type="pil", label="Upload Mask (for Inpainting)", optional=True)
73
  ],
74
  outputs=gr.Image(label="Processed Image"),
75
  title="AI-Powered Image Restoration",
 
32
  return result.images[0]
33
 
34
  def inpaint_image(base_image, mask_image):
35
+ if mask_image is None:
36
+ raise ValueError("Mask image is required for inpainting.")
37
  base_image = base_image.convert("RGB").resize((512, 512))
38
  mask_image = mask_image.convert("RGB").resize((512, 512))
39
  prompt = "A completed and visually realistic image with the missing parts filled in."
 
53
  elif task == "Denoise":
54
  return denoise_image(image)
55
  elif task == "Inpaint":
 
 
56
  return inpaint_image(image, mask)
57
  elif task == "Upscale":
58
  return upscale_image(image)
 
69
  value="Colorize"
70
  ),
71
  gr.Image(type="pil", label="Upload Image"),
72
+ gr.Image(type="pil", label="Upload Mask (for Inpainting)", value=None) # Allow empty input for mask
73
  ],
74
  outputs=gr.Image(label="Processed Image"),
75
  title="AI-Powered Image Restoration",