Test auto-mask

README.md

@@ -1,6 +1,6 @@
 ---
 title: FcF Inpainting
-emoji: 🌖
+emoji: 🚧
 colorFrom: red
 colorTo: purple
 sdk: gradio

app.py

@@ -96,25 +96,20 @@ def inpaint(input_img, mask, option):
     rgb_erased = rgb_erased * (1 - mask_tensor) # erase rgb
     rgb_erased = rgb_erased.to(torch.float32)
     
-    model = create_model("models/places_512.pkl")
-    comp_img = fcf_inpaint(G=model, org_img=rgb.to(torch.float32), erased_img=rgb_erased.to(torch.float32), mask=mask_tensor.to(torch.float32))
+    # model = create_model("models/places_512.pkl")
+    # comp_img = fcf_inpaint(G=model, org_img=rgb.to(torch.float32), erased_img=rgb_erased.to(torch.float32), mask=mask_tensor.to(torch.float32))
     rgb_erased = denorm(rgb_erased)
-    comp_img = denorm(comp_img)
+    # comp_img = denorm(comp_img)
 
-    return show_images(rgb_erased), show_images(comp_img)
+    return show_images(rgb_erased), show_images(rgb_erased)
 
 gradio_inputs = [gr.inputs.Image(type='pil',
                                  tool=None,
                                  label="Input Image"),
                 gr.inputs.Image(type='pil',source="canvas", label="Mask", invert_colors=True),
                 gr.inputs.Radio(choices=["Automatic", "Manual"], type="value", default="Manual", label="Masking Choice")
-                #  gr.inputs.Image(type='pil',
-                #                  tool=None,
-                #                  label="Mask")]
                 ]
 
-# gradio_outputs = [gr.outputs.Image(label='Auto-Detected Mask (From drawn black pixels)')]
-
 gradio_outputs = [gr.outputs.Image(label='Image with Hole'),
                  gr.outputs.Image(label='Inpainted Image')]