Update app.py

app.py

@@ -84,11 +84,11 @@ def load_pipeline(model_name):
     img_pipe.to(device)
     return pipe, img_pipe
 
-def load_img(resize_width: float,img: str):
+def load_img(resize_width,img: str):
     img = Image.open(img)
     width, height = img.size
     scale = resize_width / width
-    resize_height = height * scale
+    resize_height = int(height * scale)
     
     img = img.resize((resize_width, resize_height), Image.Resampling.LANCZOS)
     return img, resize_width, resize_height