Hugging Face's logo

Spaces:
minhdang
/
videotest
Runtime error

App Files Community
minhdang commited on
Commit
6563845
•
1 Parent(s): 8ea9aa2

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
 
36
  # Device on which to run the model
37
  # Set to cuda to load on GPU
38
- device = "cpu"
39
  model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_m16_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
 
35
 
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_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