Update app.py

app.py

@@ -27,7 +27,10 @@ def infer(video_frames, masks_frames):
 
     execute_command(command)
 
-    return "done"
+    # Get the list of files in the "results" folder
+    result_files = os.listdir(output_folder)
+
+    return result_files
 
 css="""
 #col-container{
@@ -54,7 +57,7 @@ with gr.Blocks(css=css) as demo:
                 submit_btn = gr.Button("Submit")
 
             with gr.Column():
-                result = gr.Textbox(label="Result")
+                result = gr.Files(label="Result")