model.py

from transformers import VideoMAEForVideoClassification, VideoMAEImageProcessor
from huggingface_hub import login

import os

login(os.getenv("HF_TOKEN"))



model_name = "OPear/videomae-large-finetuned-UCF-Crime"

# Load processor and model
processor = VideoMAEImageProcessor.from_pretrained(model_name)
model = VideoMAEForVideoClassification.from_pretrained(model_name)