Hugging Face's logo

Spaces:
latif98
/
latif98-videomae-base-finetuned-numbers-augmented2
Runtime error

App Files Community
latif98 commited on
Commit
853b483
•
1 Parent(s): fb872b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from torchvision.transforms import (
20
  )
21
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
22
 
23
- MODEL_CKPT = "latif98/latif98-videomae-base-finetuned-numbers-augmented2"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
 
26
  MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
 
20
  )
21
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
22
 
23
+ MODEL_CKPT = "latif98/videomae-base-finetuned-numbers-augmented2"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
 
26
  MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)