Update app.py

app.py

@@ -213,14 +213,9 @@ def run(
     use_canny: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
-    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)
-    
+    width, height  = image['composite'].size
+    ratio = np.sqrt(1024. * 1024. / (width * height))
+    new_width, new_height = int(width * ratio), int(height * ratio)
     image = image['composite'].resize((new_width, new_height))
     
     if use_canny:
@@ -261,7 +256,7 @@ def run(
         height=new_height,
     ).images[0]
     else:
-    out = pipe(
+        out = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         image=image,
@@ -270,8 +265,7 @@ 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)