Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -135,7 +135,7 @@ class ResNet18(nn.Module):
|
|
| 135 |
|
| 136 |
model = ResNet18(3, 7)
|
| 137 |
|
| 138 |
-
checkpoint = torch.load('
|
| 139 |
|
| 140 |
# The state dict will contains net.layer_name
|
| 141 |
# Our model doesn't contains `net.` so we have to rename it
|
|
|
|
| 135 |
|
| 136 |
model = ResNet18(3, 7)
|
| 137 |
|
| 138 |
+
checkpoint = torch.load('ham10000.ckpt', map_location=torch.device('cpu'))
|
| 139 |
|
| 140 |
# The state dict will contains net.layer_name
|
| 141 |
# Our model doesn't contains `net.` so we have to rename it
|