Spaces:
Runtime error
Runtime error
Update app.py
Browse filesUpdated model link
app.py
CHANGED
@@ -20,7 +20,7 @@ from torchvision.transforms import (
|
|
20 |
)
|
21 |
from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
|
22 |
|
23 |
-
MODEL_CKPT = "rickysk/
|
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/
|
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,
|