Update app.py

app.py

@@ -74,14 +74,8 @@ def generate(
     }
 
     images = model(**options).images
-    image_paths = []
-    for img in images:
-        # Save the image and create a caption (we use the filename as the caption)
-        image_path = save_image(img)
-        caption = os.path.basename(image_path)  # Use the filename as the caption
-        image_paths.append((image_path, caption))  # Add the image and caption as a tuple
-
-    return image_paths, seed
+    image_path = save_image(images[0])  # Saving the first generated image
+    return image_path
 
 with gr.Blocks(theme="soft") as demo:
     # Centered text "SNAPSCRIBE" at the top of the screen
@@ -97,12 +91,12 @@ with gr.Blocks(theme="soft") as demo:
             run_button = gr.Button("Generate Image")
             gr.Markdown("Developed using the RealVisXL_V5.0_Lightning model.", elem_id="model_info")
         with gr.Column(scale=7):
-            result = gr.Gallery(label="Generated Images", columns=2, height=1000)  # Increased height by 120%
+            result_image = gr.Image(label="Generated Image", type="filepath")
 
     run_button.click(
         fn=generate,
         inputs=[prompt],
-        outputs=[result],
+        outputs=[result_image],
     )
 
     # Footer with custom style and text