Update app.py
Browse files
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 |
-
|
| 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",
|