Hugging Face's logo

Spaces:
minhdang
/
videotest
Runtime error

App Files Community
minhdang commited on
Commit
1336378
•
1 Parent(s): 7aa2636

Update app.py

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