Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -115,7 +115,7 @@ def read_video(path, num_frames=16):
|
|
115 |
)
|
116 |
video = test_transform(frames)
|
117 |
|
118 |
-
return video
|
119 |
|
120 |
|
121 |
do_sample=False
|
|
|
115 |
)
|
116 |
video = test_transform(frames)
|
117 |
|
118 |
+
return video.permute(1, 0, 2, 3).unsqueeze(0)
|
119 |
|
120 |
|
121 |
do_sample=False
|