Spaces:
Runtime error
Runtime error
Commit
·
41f8ba0
1
Parent(s):
5f37adc
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def sample_uniform_frame_indices(clip_len, seg_len):
|
|
20 |
return np.array(indices).astype(np.int64)
|
21 |
|
22 |
def read_video_decord(file_path, indices):
|
23 |
-
vr = VideoReader(file_path, num_threads=1, ctx=gpu(0) if torch.cuda.is_available() else cpu(0))
|
24 |
video = vr.get_batch(indices).asnumpy()
|
25 |
return video
|
26 |
|
|
|
20 |
return np.array(indices).astype(np.int64)
|
21 |
|
22 |
def read_video_decord(file_path, indices):
|
23 |
+
vr = VideoReader(file_path, num_threads=1, ctx=cpu(0))#gpu(0) if torch.cuda.is_available() else cpu(0))
|
24 |
video = vr.get_batch(indices).asnumpy()
|
25 |
return video
|
26 |
|