Update app.py

app.py

@@ -46,7 +46,7 @@ def infer(video_frames, masks_frames):
     for item in bmx_trees_content:
         print(item)
 
-    return "done", bmx_trees_content[1]
+    return "done", bmx_trees_content[0], bmx_trees_content[1]
 
 css="""
 #col-container{
@@ -74,10 +74,11 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Column():
                 result = gr.Textbox(label="Result")
+                res_masked = gr.Video()
                 res_files = gr.Video()
 
         
             
-    submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_files])
+    submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_masked, res_files])
     
 demo.queue(max_size=12).launch()