Commit
•
248f463
1
Parent(s):
02c225a
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ model = InceptionResnetV1(
|
|
| 29 |
device=DEVICE
|
| 30 |
)
|
| 31 |
|
| 32 |
-
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth")
|
| 33 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 34 |
model.to(DEVICE)
|
| 35 |
model.eval()
|
|
|
|
| 29 |
device=DEVICE
|
| 30 |
)
|
| 31 |
|
| 32 |
+
checkpoint = torch.load("resnetinceptionv1_epoch_32.pth", map_location=torch.device('cpu'))
|
| 33 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 34 |
model.to(DEVICE)
|
| 35 |
model.eval()
|