Spaces:

lierzhao
/

uniformerv2_lez1

Runtime error

lierzhao commited on Apr 7, 2023

Commit

dcc27f1

1 Parent(s): 1628772

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/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')

 # 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')