Update app.py

app.py

@@ -10,9 +10,9 @@ def get_images(prompt):
     print(img_results)
     print(gallery_dir)
     
-    return img_results
+    return img_results[0]
 
 prompt_input = gr.Textbox(lines=4)
-gallery_output = gr.Gallery()
+gallery_output = gr.Image(type='filepath')
 
 gr.Interface(fn=get_images, inputs=[prompt_input], outputs=[gallery_output]).queue(max_size=10).launch(enable_queue=True)