Spaces:
Runtime error
Runtime error
Commit
•
48359f9
1
Parent(s):
4f7c4d9
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ model = InceptionResnetV1(
|
|
| 25 |
device=DEVICE
|
| 26 |
)
|
| 27 |
|
| 28 |
-
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth")
|
| 29 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 30 |
model.to(DEVICE)
|
| 31 |
model.eval()
|
|
|
|
| 25 |
device=DEVICE
|
| 26 |
)
|
| 27 |
|
| 28 |
+
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth", map_location=torch.device('cpu'))
|
| 29 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 30 |
model.to(DEVICE)
|
| 31 |
model.eval()
|