Fix size

app.py

@@ -77,8 +77,9 @@ def inpaint(input_img, mask, option):
     else:
         mask = mask.resize((width,height))
     
-    input_img = input_img.resize(512, 512)
-    mask = mask.resize(512, 512)
+    if width is not 512 or height is not 512:
+        input_img = input_img.resize((512, 512))
+        mask = mask.resize((512, 512))
 
     mask = mask.convert('L')
     mask = np.array(mask) / 255.
@@ -95,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,