lierzhao commited on
Commit
e4fbe0c
·
1 Parent(s): 46c3faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, temporal_downsample=False))
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