Update app.py

Works with any size now.

app.py

@@ -10,13 +10,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", safety_checker=None)
 pipe = pipe.to(device)
 
-def resize(height,img):
-  baseheight = height
-  img = Image.open(img)
-  hpercent = (baseheight/float(img.size[1]))
-  wsize = int((float(img.size[0])*float(hpercent)))
-  img = img.resize((wsize,baseheight), Image.Resampling.LANCZOS)
-  return img
+def resize(value,img):
+    img = Image.open(img)
+    img = img.resize((value,value))
+    return img
 
 def predict(source_img, prompt, negative_prompt):
     imageio.imwrite("data.png", source_img["image"])