Update app.py

app.py

@@ -63,14 +63,16 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
         input_image = input_image.convert("RGB")
         original_width, original_height = input_image.width, input_image.height
         now_width, now_height = input_image.width, input_image.height
-        scale = 4
-        if input_image.height // input_image.width >= 5 or input_image.width // input_image.height >= 5:
-            if input_image.height < input_image.width:
-                input_image.resize((original_width, original_height * scale ))
-                now_width, now_height = input_image.width, input_image.height * scale 
-            else:
-                input_image.resize((original_width * scale, original_height))
-                now_width, now_height = input_image.width * scale, input_image.height
+        scale = 2
+        if (input_image.height // input_image.width >= 5 or input_image.width // input_image.height >= 5) and (input_image.width < 2560 and input_image.height < 2560):
+            # if input_image.height < input_image.width:
+            #     input_image.resize((original_width, original_height * scale ))
+            #     now_width, now_height = input_image.width, input_image.height * scale 
+            # else:
+            #     input_image.resize((original_width * scale, original_height))
+            #     now_width, now_height = input_image.width * scale, input_image.height
+            input_image.resize((original_width, original_height * scale ))
+            now_width, now_height = input_image.width, input_image.height * scale 
         image = pipe(
             image=input_image, 
             prompt=prompt,