Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ import cv2
7
  import tempfile
8
  import shutil
9
  import logging
 
 
 
10
  from transformers import VideoMAEForVideoClassification, VideoMAEImageProcessor
11
  from PIL import Image
12
  from torchvision.transforms import Compose, Resize, ToTensor
 
7
  import tempfile
8
  import shutil
9
  import logging
10
+ os.environ["HF_HOME"] = "/cache"
11
+ os.environ["TRANSFORMERS_CACHE"]= "/cache"
12
+ os.environ["HF_DATASETS_CACHE"] = "/cache"
13
  from transformers import VideoMAEForVideoClassification, VideoMAEImageProcessor
14
  from PIL import Image
15
  from torchvision.transforms import Compose, Resize, ToTensor