Spaces:
Sleeping
Sleeping
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
|