Update app.py

app.py

@@ -213,9 +213,14 @@ def run(
     use_canny: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
-    width, height  = image['composite'].size
-    ratio = np.sqrt(1024. * 1024. / (width * height))
-    new_width, new_height = int(width * ratio), int(height * ratio)
+    width, height = image['composite'].size
+    
+    # Calculate new dimensions while maintaining aspect ratio
+    max_size = 1024
+    ratio = min(max_size / width, max_size / height)
+    new_width = int(width * ratio)
+    new_height = int(height * ratio)
+    
     image = image['composite'].resize((new_width, new_height))
     
     if use_canny:
@@ -256,7 +261,7 @@ def run(
         height=new_height,
     ).images[0]
     else:
-        out = pipe(
+    out = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         image=image,
@@ -265,7 +270,8 @@ def run(
         controlnet_conditioning_scale=controlnet_conditioning_scale,
         guidance_scale=guidance_scale,
         width=new_width,
-        height=new_height,).images[0]
+        height=new_height,
+    ).images[0]
 
     return (controlnet_img, out)