IbrahimHasani commited on
Commit
41f8ba0
·
1 Parent(s): 5f37adc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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