change device from 'cuda' to 'cpu'
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from train import ASR_Model
|
|
| 6 |
from model_cnn import Model
|
| 7 |
|
| 8 |
def pre(audio):
|
| 9 |
-
model = ASR_Model(device='
|
| 10 |
|
| 11 |
result = model.predict(audio)
|
| 12 |
s = ''
|
|
|
|
| 6 |
from model_cnn import Model
|
| 7 |
|
| 8 |
def pre(audio):
|
| 9 |
+
model = ASR_Model(device='cpu',model_path='model/model.pth',pinyin_path ='pinyin.txt')
|
| 10 |
|
| 11 |
result = model.predict(audio)
|
| 12 |
s = ''
|