Spaces:
Runtime error
Runtime error
jwyang
commited on
Commit
·
2f7472f
1
Parent(s):
755dbb1
change map location
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ build model
|
|
| 68 |
model = build_model(config)
|
| 69 |
|
| 70 |
url = './in21k_yfcc14m_gcc15m_swin_base.pth'
|
| 71 |
-
checkpoint = torch.load(url)
|
| 72 |
model.load_state_dict(checkpoint["model"])
|
| 73 |
model.eval()
|
| 74 |
|
|
|
|
| 68 |
model = build_model(config)
|
| 69 |
|
| 70 |
url = './in21k_yfcc14m_gcc15m_swin_base.pth'
|
| 71 |
+
checkpoint = torch.load(url, map_location="cpu")
|
| 72 |
model.load_state_dict(checkpoint["model"])
|
| 73 |
model.eval()
|
| 74 |
|