Spaces:
Runtime error
Runtime error
Commit
•
7f2b443
1
Parent(s):
0ffc184
Update app.py
Browse files
app.py
CHANGED
|
@@ -109,7 +109,7 @@ model = LightningModel()
|
|
| 109 |
run_name = "wav2vec"
|
| 110 |
|
| 111 |
checkpoint_path = "./wav2vec-epoch=epoch=4.ckpt.ckpt"
|
| 112 |
-
checkpoint = torch.load(checkpoint_path)
|
| 113 |
model.load_state_dict(checkpoint['state_dict'])
|
| 114 |
trainer = Trainer()
|
| 115 |
|
|
|
|
| 109 |
run_name = "wav2vec"
|
| 110 |
|
| 111 |
checkpoint_path = "./wav2vec-epoch=epoch=4.ckpt.ckpt"
|
| 112 |
+
checkpoint = torch.load(checkpoint_path,map_location=torch.device('cpu'))
|
| 113 |
model.load_state_dict(checkpoint['state_dict'])
|
| 114 |
trainer = Trainer()
|
| 115 |
|