Spaces:
Running
Running
Seventh commit
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ class LabelClassifier(nn.Module):
|
|
| 62 |
return out
|
| 63 |
|
| 64 |
model = LabelClassifier().to(device)
|
| 65 |
-
model.load_state_dict(torch.load('classifier.pth'))
|
| 66 |
|
| 67 |
|
| 68 |
|
|
|
|
| 62 |
return out
|
| 63 |
|
| 64 |
model = LabelClassifier().to(device)
|
| 65 |
+
model.load_state_dict(torch.load('classifier.pth', map_location=torch.device('cpu')))
|
| 66 |
|
| 67 |
|
| 68 |
|