Spaces:

Alesteba
/

AP_02

Sleeping

Alesteba commited on Apr 17, 2023

Commit

970f03e

1 Parent(s): 6feb878

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ model_2 = models.torchvision.retinanet.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('./retinanet_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')
 model_2.load_state_dict(state_dict)