Hugging Face's logo

Spaces:
minhdang
/
videotest
Runtime error

App Files Community
minhdang commited on
Commit
e046e9c
·
verified ·
1 Parent(s): 90cf636

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ from huggingface_hub import hf_hub_download
35
  # Device on which to run the model
36
  # Set to cuda to load on GPU
37
  device = "cuda"
38
- model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_k400_f16_res224.pth")
39
  model_image_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_in1k_res224.pth")
40
  # Pick a pretrained model
41
  model_video = videomamba_tiny(num_classes=400, num_frames=16)
 
35
  # Device on which to run the model
36
  # Set to cuda to load on GPU
37
  device = "cuda"
38
+ model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_k400_f32_res224.pth")
39
  model_image_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_in1k_res224.pth")
40
  # Pick a pretrained model
41
  model_video = videomamba_tiny(num_classes=400, num_frames=16)