Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ class Uniformerv2(nn.Module):
|
|
31 |
device = "cpu"
|
32 |
model_path = hf_hub_download(repo_id="lierzhao/mit_uniformerv2_l14_8x336.pyth", filename="mit_uniformerv2_l14_8x336.pyth")
|
33 |
# Pick a pretrained model
|
34 |
-
model = Uniformerv2(uniformerv2_l14_336(pretrained=False, t_size=8, no_lmhra=True
|
35 |
state_dict = torch.load(model_path, map_location='cpu')
|
36 |
model.load_state_dict(state_dict)
|
37 |
|
|
|
31 |
device = "cpu"
|
32 |
model_path = hf_hub_download(repo_id="lierzhao/mit_uniformerv2_l14_8x336.pyth", filename="mit_uniformerv2_l14_8x336.pyth")
|
33 |
# Pick a pretrained model
|
34 |
+
model = Uniformerv2(uniformerv2_l14_336(pretrained=False, t_size=8, no_lmhra=True))
|
35 |
state_dict = torch.load(model_path, map_location='cpu')
|
36 |
model.load_state_dict(state_dict)
|
37 |
|