Update app.py

app.py

@@ -46,12 +46,12 @@ def resize_image_dimensions(
     w_original, h_original = w, h
  
     if w > h:
-         aspect_ratiow = w / h
+         aspect_ratiow = h / w
          w = IMAGE_SIZE
          h = IMAGE_SIZE * aspect_ratiow 
         
     if h > w:
-         aspect_ratioh = h / w
+         aspect_ratioh = w / h
          w = IMAGE_SIZE * aspect_ratioh
          h = IMAGE_SIZE 
         
@@ -79,7 +79,7 @@ def resize_image_dimensions(
     # new_width = new_width - (new_width % 32)
     # new_height = new_height - (new_height % 32)
 
-    return int(h), int(w)
+    return int(w), int(h)
 
 
 @spaces.GPU(duration=80)
@@ -100,7 +100,7 @@ def process(
 
     if not image:
         gr.Info("Please upload an image.")
-        return None, None
+        return None# , None
         
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
     resized_image = image.resize((width, height), Image.LANCZOS)
@@ -110,6 +110,7 @@ def process(
  
     # if not mask:
        # gr.Info("Please draw a mask on the image.")
+    
     pipe2.load_lora_weights("SIGMitch/KIT")
     result = pipe2(
         prompt=input_text,
@@ -125,7 +126,7 @@ def process(
         )
     print('INFERENCE DONE')
     #return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
-    return result.images[0], None #result.images[1]
+    return result.images[0]# , None #result.images[1]
 
      #resized_mask = mask.resize((width, height), Image.LANCZOS)
 
@@ -204,8 +205,8 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_image_component = gr.Image(
                 type='pil', image_mode='RGB', label='Generated image', format="png")
-            output_image_component2 = gr.Image(
-                type='pil', image_mode='RGB', label='Generated image', format="png")
+            # output_image_component2 = gr.Image(
+            #     type='pil', image_mode='RGB', label='Generated image', format="png")
 
 
 
@@ -221,8 +222,8 @@ with gr.Blocks() as demo:
             lorasteps
         ],
         outputs=[
-            output_image_component,
-            output_image_component2
+            output_image_component# ,
+            # output_image_component2
         ]
     )