Update app.py

app.py

@@ -18,13 +18,14 @@ def infer(prompt,image):
     print(result[0])
     print(result[0][0])
     # return "done"
-    return [os.path.join(result[0], img) for img in os.listdir(result[0])]
+    images = [os.path.join(result[0], img) for img in os.listdir(result[0])]
+    return images[0]
 
 with gr.Blocks() as demo:
     with gr.Column():
         prompt = gr.Textbox(placeholder="enter prompt")
         image = gr.Image(source="upload", type="filepath")
-        gallery_out = gr.Gallery()
+        image_out = gr.Image()
         #text_out = gr.Textbox()
         submit_btn = gr.Button("go")
 
@@ -32,7 +33,7 @@ with gr.Blocks() as demo:
         prompt,
         image
     ]
-    outputs = [gallery_out]
+    outputs = [image_out]
     #outputs = [text_out]
     submit_btn.click(infer, inputs, outputs)