fix bug

deploy_model.py

@@ -2,7 +2,7 @@
 Author: wuxulong19950206 [email protected]
 Date: 2024-03-11 19:46:42
 LastEditors: wuxulong19950206 [email protected]
-LastEditTime: 2024-03-11 20:29:07
+LastEditTime: 2024-03-11 20:57:50
 FilePath: \apeech_rec\deploy_model.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -29,7 +29,10 @@ def get_pretrained_model():
     
     feat = Spectrogram()
     ms = ModelSpeech(sm251bn, feat, max_label_length=64)
-
+    if os.path.exists('models\SpeechModel251bn_epoch12.model.h5'):
+        print('*[提示] 加载预训练的声学模型')
+    else:
+        print('*[提示] 预训练的声学模型不存在')
     ms.load_model('models\SpeechModel251bn_epoch12.model.h5')
     return ms
 def decode(model,filename):