Spaces:

Sense-X
/

uniformer_video_demo

Runtime error

Andy1621 commited on Feb 9, 2022

Commit

fa4b7a9

•

1 Parent(s): 0599b47

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ def inference(video):
 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cpu"
-model_path = hf_hub_download(repo_id="Andy1621/uniformer", filename="uniformer_small_k400_16x8.pth")
 # Pick a pretrained model
 model = uniformer_small()
 state_dict = torch.load(model_path, map_location='cpu')

 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cpu"
+model_path = hf_hub_download(repo_id="Sense-X/uniformer_video", filename="uniformer_small_k400_16x8.pth")
 # Pick a pretrained model
 model = uniformer_small()
 state_dict = torch.load(model_path, map_location='cpu')