Update app.py
Browse files
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)",
|
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",
|