Spaces:
Build error
Build error
quick fix
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ SAMPLE_RATE = 16000
|
|
| 11 |
AUDIO_LEN = 2.90
|
| 12 |
|
| 13 |
model = HarmonicCNN()
|
| 14 |
-
S = torch.load('models/best_model.pth')
|
| 15 |
model.load_state_dict(S)
|
| 16 |
|
| 17 |
LABELS = [
|
|
|
|
| 11 |
AUDIO_LEN = 2.90
|
| 12 |
|
| 13 |
model = HarmonicCNN()
|
| 14 |
+
S = torch.load('models/best_model.pth', map_location=torch.device('cpu'))
|
| 15 |
model.load_state_dict(S)
|
| 16 |
|
| 17 |
LABELS = [
|