Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -115,11 +115,11 @@ def read_video(path, num_frames=16):
|
|
115 |
)
|
116 |
video = test_transform(frames)
|
117 |
|
118 |
-
return video
|
119 |
|
120 |
|
121 |
do_sample=False
|
122 |
-
num_beams=
|
123 |
max_length=30
|
124 |
|
125 |
|
|
|
115 |
)
|
116 |
video = test_transform(frames)
|
117 |
|
118 |
+
return video.unsqueeze(0)
|
119 |
|
120 |
|
121 |
do_sample=False
|
122 |
+
num_beams=5
|
123 |
max_length=30
|
124 |
|
125 |
|