Update app.py

app.py

@@ -17,23 +17,23 @@ def infer(prompt,image):
     print(result)
     print(result[0])
     print(result[0][0])
-    return "done"
-    #return [os.path.join(result[0], img) for img in os.listdir(result[0])]
+    # return "done"
+    return [os.path.join(result[0], img) for img in os.listdir(result[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()
-        text_out = gr.Textbox()
+        gallery_out = gr.Gallery()
+        #text_out = gr.Textbox()
         submit_btn = gr.Button("go")
 
     inputs = [
         prompt,
         image
     ]
-    #outputs = [gallery_out]
-    
-    submit_btn.click(infer, inputs, text_out)
+    outputs = [gallery_out]
+    outputs = [text_out]
+    submit_btn.click(infer, inputs, outputs)
 
 demo.launch().queue()