Spaces:

Alesteba
/

AP_02

Sleeping

Alesteba commited on Apr 18, 2023

Commit

e3c5df3

1 Parent(s): 9f1ce46

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,8 +20,8 @@ model_2 = model_type.model(
 # load from model_repo:
-# 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'))

 # load from model_repo:
+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'))