Update app.py

app.py

@@ -41,8 +41,28 @@ def resize_image_dimensions(
     original_resolution_wh: Tuple[int, int],
     maximum_dimension: int = IMAGE_SIZE
 ) -> Tuple[int, int]:
-    width, height = original_resolution_wh
-
+   # width, height = original_resolution_wh
+    w, h = original_resolution_wh
+    w_original, h_original = w, h
+ 
+    if w > h:
+         aspect_ratiow = w / h
+         w = IMAGE_SIZE
+         h = IMAGE_SIZE * aspect_ratiow 
+        
+    if h > w:
+         aspect_ratioh = h / w
+         w = IMAGE_SIZE * aspect_ratioh
+         h = IMAGE_SIZE 
+        
+    if h == w:
+         w = IMAGE_SIZE
+         h = IMAGE_SIZE   
+    
+    #was_resized = False
+
+     #w = w - w % 8
+     #h = h - h % 8
      #if width <= maximum_dimension and height <= maximum_dimension:
      #    width = width - (width % 32)
       #   height = height - (height % 32)
@@ -59,7 +79,7 @@ def resize_image_dimensions(
     # new_width = new_width - (new_width % 32)
     # new_height = new_height - (new_height % 32)
 
-    return 1024, 1024
+    return w, h
 
 
 @spaces.GPU(duration=80)