fffiloni commited on
Commit
a77a510
1 Parent(s): ca06c20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -46,7 +46,7 @@ def infer(video_frames, masks_frames):
46
  for item in bmx_trees_content:
47
  print(item)
48
 
49
- return "done", bmx_trees_content[1]
50
 
51
  css="""
52
  #col-container{
@@ -74,10 +74,11 @@ with gr.Blocks(css=css) as demo:
74
 
75
  with gr.Column():
76
  result = gr.Textbox(label="Result")
 
77
  res_files = gr.Video()
78
 
79
 
80
 
81
- submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_files])
82
 
83
  demo.queue(max_size=12).launch()
 
46
  for item in bmx_trees_content:
47
  print(item)
48
 
49
+ return "done", bmx_trees_content[0], bmx_trees_content[1]
50
 
51
  css="""
52
  #col-container{
 
74
 
75
  with gr.Column():
76
  result = gr.Textbox(label="Result")
77
+ res_masked = gr.Video()
78
  res_files = gr.Video()
79
 
80
 
81
 
82
+ submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result, res_masked, res_files])
83
 
84
  demo.queue(max_size=12).launch()