Update app.py

app.py

@@ -14,9 +14,11 @@ def infer(prompt,image):
         512,
         0,
         fn_index=0)
-
-    print(os.listdir(result))
-    return [os.path.join(result[0], img) for img in os.listdir(result[0])]
+    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])]
 
 with gr.Blocks() as demo:
     with gr.Column():
@@ -31,6 +33,6 @@ with gr.Blocks() as demo:
     ]
     outputs = [gallery_out]
     
-    submit_btn.click(infer, inputs, outputs)
+    submit_btn.click(infer, inputs, text)
 
 demo.launch().queue()