Update app.py

app.py

@@ -73,14 +73,14 @@ def process(
 ):
     if not input_text:
         gr.Info("Please enter a text prompt.")
-        return None, None, None, None
+        return None, None
     input_text = "A military COR3 "+input_text
     image = input_image_editor['background']
     mask = input_image_editor['layers'][0]
 
     if not image:
         gr.Info("Please upload an image.")
-        return None, None, None, None
+        return None, None
         
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
     resized_image = image.resize((width, height), Image.LANCZOS)
@@ -96,14 +96,14 @@ def process(
         image=resized_image,
         width=width,
         height=height,
-        num_images_per_prompt =4,
+        num_images_per_prompt =2,
         strength=strength_slider,
         generator=generator,
         joint_attention_kwargs={"scale": 1.2},
         num_inference_steps=num_inference_steps_slider
         )
     print('INFERENCE DONE')
-    return result.images[0], result.images[1], result.images[2], result.images[3]
+    return result.images[0], result.images[1]
 
      #resized_mask = mask.resize((width, height), Image.LANCZOS)
 
@@ -183,10 +183,7 @@ with gr.Blocks() as demo:
                 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_component3 = gr.Image(
-                type='pil', image_mode='RGB', label='Generated image', format="png")
-            output_image_component4 = gr.Image(
-                type='pil', image_mode='RGB', label='Generated image', format="png")
+
 
 
     submit_button_component.click(
@@ -201,9 +198,7 @@ with gr.Blocks() as demo:
         ],
         outputs=[
             output_image_component,
-            output_image_component2,
-            output_image_component3,
-            output_image_component4
+            output_image_component2
         ]
     )