Spaces:

jw2yang
/

unicl-img-recog-demo

Runtime error

jwyang commited on Jun 4, 2022

Commit

69ddcf7

1 Parent(s): c7077ad

minor change to app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ build model
 model = build_model(config)
 url = './in21k_yfcc14m_gcc15m_swin_base.pth'
-checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", check_hash=True)
 model.load_state_dict(checkpoint["model"])
 model.eval()

 model = build_model(config)
 url = './in21k_yfcc14m_gcc15m_swin_base.pth'
+checkpoint = torch.load(url)
 model.load_state_dict(checkpoint["model"])
 model.eval()