Update app.py

app.py

@@ -41,13 +41,13 @@ def infer(video_frames, masks_frames):
     # List the content of the "bmx-trees" folder within "results"
     bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
     bmx_trees_content = os.listdir(bmx_trees_folder)
+    bmx_trees_content = [os.path.join(bmx_trees_folder, item) for item in os.listdir(bmx_trees_folder)]
 
     print(f"Contents of the {bmx_trees_folder} folder:")
     for item in bmx_trees_content:
         print(item)
 
-
-    return "done"
+    return "done", bmx_tress_content
 
 css="""
 #col-container{
@@ -75,9 +75,10 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Column():
                 result = gr.Textbox(label="Result")
+                res_files = gr.Files()
 
         
             
-    submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result])
+    submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_files])
     
 demo.queue(max_size=12).launch()