Spaces:

archit11
/

videomae-base-finetuned-ucfcrime-full

Sleeping

archit11 commited on Mar 18

Commit

dd8e993

•

1 Parent(s): 8552f78

new model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ from torchvision.transforms import (
 )
 from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
-MODEL_CKPT = "archit11/videomae-base-finetuned-fight-nofight-subset2"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)

 )
 from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
+MODEL_CKPT = "archit11/videomae-base-finetuned-ucfcrime-full"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)