Update app.py

app.py

@@ -15,7 +15,7 @@ def infer(prompt,image):
         fn_index=0)
 
     print(result)
-    return result[0]
+    return [os.path.join(result[0], img) for img in os.listdir(result[0])]
 
 with gr.Blocks() as demo:
     with gr.Column():