Spaces:

mayhug
/

rf5-anime-image-label

Runtime error

mayhug commited on Feb 3, 2022

Commit

d584e3b

1 Parent(s): 1fe5e1e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,6 @@ torch.hub.load_state_dict_from_url = load_state_dict_from_url
 model = torch.hub.load(
     "RF5/danbooru-pretrained",
     "resnet50",
-    map_location="cpu",
 )
 model.eval()

 model = torch.hub.load(
     "RF5/danbooru-pretrained",
     "resnet50",
 )
 model.eval()