Spaces:

Alesteba
/

AP_02

Sleeping

Alesteba commited on Apr 17, 2023

Commit

df5a255

1 Parent(s): 865fc80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ model_2 = model_type.model(
 # from huggingface_hub import hf_hub_download
 # hf_hub_download(repo_id="Alesteba/deep_model_02", filename="retinanet_racoon.pth")
-state_dict = torch.load('./mmdet_racoon.pth', map_location=torch.device('cpu'))
 model_2.load_state_dict(state_dict)

 # from huggingface_hub import hf_hub_download
 # hf_hub_download(repo_id="Alesteba/deep_model_02", filename="retinanet_racoon.pth")
+state_dict = torch.load('./retinanet_racoon.pth', map_location=torch.device('cpu'))
 model_2.load_state_dict(state_dict)