Update test.py
Browse files
test.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from train import ASR_Model
|
| 2 |
from model_cnn import Model
|
| 3 |
|
| 4 |
-
model = ASR_Model(device='cuda',model_path='model/model.pth')
|
| 5 |
result = model.predict('Examples_中原石化加油站.wav')
|
| 6 |
print(result)
|
|
|
|
| 1 |
from train import ASR_Model
|
| 2 |
from model_cnn import Model
|
| 3 |
|
| 4 |
+
model = ASR_Model(device='cuda',model_path='model/model.pth',pinyin_path ='pinyin.txt')
|
| 5 |
result = model.predict('Examples_中原石化加油站.wav')
|
| 6 |
print(result)
|