Update app.py (#8)

- Update app.py (53dfd2c355f763f71efe81291f409a71d46487a3)


Co-authored-by: Harsh Pahwa <[email protected]>

app.py

@@ -123,15 +123,10 @@ def predict(
     size1, size2 = input_image["image"].convert("RGB").size
 
     if task != "image-outpainting":
-        if size1 < size2:
-            input_image["image"] = input_image["image"].convert("RGB").resize((640, int(size2 / size1 * 640)))
-        else:
-            input_image["image"] = input_image["image"].convert("RGB").resize((int(size1 / size2 * 640), 640))
+        input_image["image"] = input_image["image"].convert("RGB").resize((1024, 1024), Image.LANCZOS)
     else:
-        if size1 < size2:
-            input_image["image"] = input_image["image"].convert("RGB").resize((512, int(size2 / size1 * 512)))
-        else:
-            input_image["image"] = input_image["image"].convert("RGB").resize((int(size1 / size2 * 512), 512))
+        input_image["image"] = input_image["image"].convert("RGB").resize((1024, 1024), Image.LANCZOS)
+        
 
     if task == "image-outpainting" or task == "context-aware":
         prompt = prompt + " empty scene"
@@ -171,8 +166,8 @@ def predict(
 
     W = int(np.shape(img)[0] - np.shape(img)[0] % 8)
     H = int(np.shape(img)[1] - np.shape(img)[1] % 8)
-    input_image["image"] = input_image["image"].resize((H, W))
-    input_image["mask"] = input_image["mask"].resize((H, W))
+    input_image["image"] = input_image["image"].resize((H, W), Image.LANCZOS)
+    input_image["mask"] = input_image["mask"].resize((H, W), Image.LANCZOS)
 
     np_inpimg = np.array(input_image["image"])
     np_inmask = np.array(input_image["mask"]) / 255.0