Alesteba commited on
Commit
2c487ac
·
1 Parent(s): ff83a4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,10 +18,10 @@ model_2 = models.torchvision.retinanet.model(
18
 
19
  # load from model_repo:
20
 
21
- from huggingface_hub import hf_hub_download
22
- hf_hub_download(repo_id="Alesteba/deep_model_02", filename="retinanet_racoon.pth")
23
 
24
- state_dict = torch.load('retinanet_racoon.pth', map_location=torch.device('cpu'))
25
 
26
  model_2.load_state_dict(state_dict)
27
 
 
18
 
19
  # load from model_repo:
20
 
21
+ # from huggingface_hub import hf_hub_download
22
+ # hf_hub_download(repo_id="Alesteba/deep_model_02", filename="retinanet_racoon.pth")
23
 
24
+ state_dict = torch.load('./retinanet_racoon.pth', map_location=torch.device('cpu'))
25
 
26
  model_2.load_state_dict(state_dict)
27