innat commited on
Commit
9ffae4a
1 Parent(s): 2a42037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -101,7 +101,7 @@ def inference(video_file, data_type, mask_ratio):
101
 
102
  combined_gif = 'combined.gif'
103
  imageio.mimsave(combined_gif, frames, duration=300, loop=0)
104
- return confidences
105
 
106
 
107
  def main():
@@ -132,7 +132,7 @@ def main():
132
  ],
133
  outputs=[
134
  gr.Label(num_top_classes=3, label='scores'),
135
- # gr.Image(type="filepath", label='reconstructed')
136
  ],
137
  examples=sample_example,
138
  title="VideoMAE",
 
101
 
102
  combined_gif = 'combined.gif'
103
  imageio.mimsave(combined_gif, frames, duration=300, loop=0)
104
+ return confidences, combined_gif
105
 
106
 
107
  def main():
 
132
  ],
133
  outputs=[
134
  gr.Label(num_top_classes=3, label='scores'),
135
+ gr.Image(type="filepath", label='reconstructed')
136
  ],
137
  examples=sample_example,
138
  title="VideoMAE",