jpterry commited on
Commit
d7f5ea2
·
1 Parent(s): b7faf40
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -292,10 +292,10 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
292
  stochastic_depth_prob=hparams.stochastic_depth_prob,
293
  width_mult=hparams.width_mult,
294
  depth_mult=hparams.depth_mult,)
295
- model_url = cached_download(hf_hub_url(model_loading_name, filename="pytorch_model.bin"))
296
  # print(model_url)
297
 
298
- loaded = torch.load(model_url, map_location='cpu')
299
  print(loaded)
300
 
301
  model.load_state_dict(loaded)
 
292
  stochastic_depth_prob=hparams.stochastic_depth_prob,
293
  width_mult=hparams.width_mult,
294
  depth_mult=hparams.depth_mult,)
295
+ model_url = cached_download(hf_hub_url(model_path + model_loading_name, filename="pytorch_model.bin"))
296
  # print(model_url)
297
 
298
+ loaded = torch.load(model_url, map_location='cpu', )
299
  print(loaded)
300
 
301
  model.load_state_dict(loaded)