Update app.py

app.py

@@ -205,7 +205,7 @@ def infer(
     original_image = input_image.convert("RGB")
     
     # Use the new function to update dimensions
-    width, height = original_image.size
+    width, height = update_dimensions_on_upload(original_image)
 
     result = pipe(
         image=original_image,