rickysk commited on
Commit
e9fa08b
1 Parent(s): dccaadc

Update app.py

Browse files

Updated model link

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ from torchvision.transforms import (
20
  )
21
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
22
 
23
- MODEL_CKPT = "rickysk/rickysk-videomae-base-ipm_all_videos"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
 
26
  MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
@@ -133,7 +133,7 @@ gr.Interface(
133
  ),
134
  article=(
135
  "<div style='text-align: center;'><a href='https://huggingface.co/docs/transformers/model_doc/videomae' target='_blank'>VideoMAE</a>"
136
- " <center><a href='https://huggingface.co/rickysk/rickysk-videomae-base-ipm_all_videos' target='_blank'>Fine-tuned Model</a></center></div>"
137
  ),
138
  allow_flagging=False,
139
  allow_screenshot=False,
 
20
  )
21
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
22
 
23
+ MODEL_CKPT = "rickysk/videomae-base-ipm_all_videos"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
 
26
  MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
 
133
  ),
134
  article=(
135
  "<div style='text-align: center;'><a href='https://huggingface.co/docs/transformers/model_doc/videomae' target='_blank'>VideoMAE</a>"
136
+ " <center><a href='https://huggingface.co/rickysk/videomae-base-ipm_all_videos' target='_blank'>Fine-tuned Model</a></center></div>"
137
  ),
138
  allow_flagging=False,
139
  allow_screenshot=False,