Spaces:
Sleeping
Sleeping
Commit ·
28a264f
1
Parent(s): b73f80f
fix weight file name import in app.py file
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ params = {
|
|
| 22 |
}
|
| 23 |
print('testesesesf')
|
| 24 |
model = CNN(**params)
|
| 25 |
-
model.load_state_dict(torch.load('model_weights.
|
| 26 |
model.eval()
|
| 27 |
|
| 28 |
# utils
|
|
|
|
| 22 |
}
|
| 23 |
print('testesesesf')
|
| 24 |
model = CNN(**params)
|
| 25 |
+
model.load_state_dict(torch.load('model_weights.pth', map_location='cpu'))
|
| 26 |
model.eval()
|
| 27 |
|
| 28 |
# utils
|