tanthinhdt commited on
Commit
9cd4a53
·
verified ·
1 Parent(s): c8d2e6a

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ examples = [
23
 
24
  # Initialize the model and image processor.
25
  device = 'cpu'
26
- model_name = 'VieSignLang/videomae_skeleton_v2.3'
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)