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",
|