Update app.py

app.py

@@ -26,7 +26,7 @@ def generate_images(text, selected_model):
         return "Invalid model selection."
     
     results = []
-    for i in range(3):
+    for i in range(2):
         modified_text = f"{text} variation {i+1}"
         result = model(modified_text)
         results.append(result)
@@ -46,7 +46,7 @@ interface = gr.Interface(
     outputs=[
         gr.Image(label="Generated Image 1"),
         gr.Image(label="Generated Image 2"),
-        gr.Image(label="Generated Image 3"),
+        # gr.Image(label="Generated Image 3"),
     ],
     theme="Yntec/HaleyCH_Theme_Orange",
     description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",