Spaces:

lierzhao
/

uniformerv2_lez1

Runtime error

lierzhao commited on Apr 7, 2023

Commit

1628772

1 Parent(s): 2da76e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ class Uniformerv2(nn.Module):
 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cpu"
-model_path = hf_hub_download(repo_id="lierzhao/uniformerv2_lez1", filename="lierzhao/mit_uniformerv2_l14_8x336.pyth/mit_uniformerv2_l14_8x336.pyth")
 # Pick a pretrained model
 model = Uniformerv2(uniformerv2_b16(pretrained=False, t_size=8, no_lmhra=True, temporal_downsample=False))
 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="lierzhao/mit_uniformerv2_l14_8x336.pyth/", filename="mit_uniformerv2_l14_8x336.pyth")
 # Pick a pretrained model
 model = Uniformerv2(uniformerv2_b16(pretrained=False, t_size=8, no_lmhra=True, temporal_downsample=False))
 state_dict = torch.load(model_path, map_location='cpu')