Hugging Face's logo

Spaces:
omermazig
/
videomae-finetuned-nba-5-class-multilabel
Runtime error

App Files Community
omermazig commited on
Commit
d0339fe
•
1 Parent(s): d4a9f94

Update app.py

Browse files

Fix model name

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from transformers import VideoMAEForVideoClassification, VideoMAEFeatureExtracto
20
 
21
  from video_utils import change_video_resolution_and_fps
22
 
23
- MODEL_CKPT = "omermazig/videomae-finetuned-nba-5-class-4-batch-8000-vid-multilabel-4"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  CLIPS_FROM_SINGLE_VIDEO = 5
26
 
 
20
 
21
  from video_utils import change_video_resolution_and_fps
22
 
23
+ MODEL_CKPT = "omermazig/videomae-finetuned-nba-5-class-4-batch-8000-vid-multilabel"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  CLIPS_FROM_SINGLE_VIDEO = 5
26