Spaces:
Build error
Build error
Upload app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ examples = [
|
|
| 23 |
|
| 24 |
# Initialize the model and image processor.
|
| 25 |
device = 'cpu'
|
| 26 |
-
model_name = 'VieSignLang/
|
| 27 |
image_processor = VideoMAEImageProcessor.from_pretrained(model_name)
|
| 28 |
model = VideoMAEForVideoClassification.from_pretrained(model_name)
|
| 29 |
model = model.eval().to(device)
|
|
|
|
| 23 |
|
| 24 |
# Initialize the model and image processor.
|
| 25 |
device = 'cpu'
|
| 26 |
+
model_name = 'VieSignLang/videomae_skeleton_v1.0'
|
| 27 |
image_processor = VideoMAEImageProcessor.from_pretrained(model_name)
|
| 28 |
model = VideoMAEForVideoClassification.from_pretrained(model_name)
|
| 29 |
model = model.eval().to(device)
|