Kaushik066 commited on
Commit
184cb45
·
1 Parent(s): bc40b6c

adding additional lib

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -7,6 +7,11 @@ from transformers import VivitImageProcessor, VivitConfig, VivitModel
7
  from torch.utils.data import Dataset, DataLoader
8
  # For GPU
9
  from accelerate import Accelerator, notebook_launcher
 
 
 
 
 
10
  # General Libraries
11
  import os
12
  import PIL
 
7
  from torch.utils.data import Dataset, DataLoader
8
  # For GPU
9
  from accelerate import Accelerator, notebook_launcher
10
+ # Use PyTorch bridge for Decord
11
+ import decord
12
+ from decord.bridge import set_bridge
13
+ decord.bridge.set_bridge("torch")
14
+ from decord import VideoReader
15
  # General Libraries
16
  import os
17
  import PIL