Update app.py

app.py

@@ -17,13 +17,13 @@ def generate_image(prompt, negative_prompt="Low quality", width=512, height=512)
         image.save(f"squirrel-{idx}.png")
     end_time = time.time()
     elapsed_time = round(end_time - start_time, 2)
-    return f"Images generated successfully! (Elapsed Time: {elapsed_time} seconds)"
+    return resized_images[0]
 
 # Define the interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=["text", "text", "number", "number"],
-    outputs=gr.outputs.Image(label="Generated Image"),
+    outputs=gr.outputs.Image(type="pil", label="Generated Image"),
     layout="vertical",
     title="Image Generation",
     description="Generate images based on prompts.",
@@ -32,4 +32,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()