Finalize demo

README.md

@@ -1,7 +1,7 @@
 ---
 title: FcF Inpainting
-emoji: 🚧
-colorFrom: red
+emoji: 🪄 ✨ ✨
+colorFrom: blue
 colorTo: purple
 sdk: gradio
 sdk_version: 2.9.4

app.py

@@ -96,12 +96,12 @@ 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(rgb_erased)
+    return show_images(rgb_erased), show_images(comp_img)
 
 gradio_inputs = [gr.inputs.Image(type='pil',
                                  tool=None,