Alesteba commited on
Commit
45d2fe3
·
1 Parent(s): e3c5df3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,9 +21,9 @@ model_2 = model_type.model(
21
  # load from model_repo:
22
 
23
  from huggingface_hub import hf_hub_download
24
- hf_hub_download(repo_id="Alesteba/deep_model_02", filename="retinanet_racoon.pth")
25
 
26
- state_dict = torch.load('./retinanet_racoon.pth', map_location=torch.device('cpu'))
27
 
28
  model_2.load_state_dict(state_dict)
29
 
 
21
  # load from model_repo:
22
 
23
  from huggingface_hub import hf_hub_download
24
+ hf_hub_download(repo_id="Alesteba/deep_model_02", filename="ross_racoon.pth")
25
 
26
+ state_dict = torch.load('./ross_racoon.pth', map_location=torch.device('cpu'))
27
 
28
  model_2.load_state_dict(state_dict)
29