Update app.py (#6)

- Update app.py (93fff84a1c1009e353e918ccdcd7c5bc78744732)


Co-authored-by: Kattamuri Tejo Vardhan <tejovk311@users.noreply.huggingface.co>

app.py

@@ -7,12 +7,10 @@ import cv2
 import tempfile
 import shutil
 import logging
-os.environ["HF_HOME"]          = "/cache"
-os.environ["TRANSFORMERS_CACHE"]= "/cache"
-os.environ["HF_DATASETS_CACHE"] = "/cache"
 from transformers import VideoMAEForVideoClassification, VideoMAEImageProcessor
 from PIL import Image
 from torchvision.transforms import Compose, Resize, ToTensor
+os.makedirs("./.cache", exist_ok=True)
 
 app = Flask(__name__)
 # Configure logging
@@ -31,7 +29,7 @@ def load_model():
     if model is None:
         model_name = "OPear/videomae-large-finetuned-UCF-Crime"
         logger.info(f"Loading model {model_name} on {device}...")
-        model = VideoMAEForVideoClassification.from_pretrained(model_name).to(device)
+        model = VideoMAEForVideoClassification.from_pretrained("OPear/videomae-large-finetuned-UCF-Crime",cache_dir="./.cache").to(device)
         processor = VideoMAEImageProcessor.from_pretrained(model_name)
         transform = Compose([
             Resize((224, 224)),