Update text_audio_emo.py

1933211129 · Apr 10, 2024 · ae6d778 · ae6d778
1 parent 47b5a80
commit ae6d778
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/speech_ai/login/py/EGG/text_audio_emo.py b/speech_ai/login/py/EGG/text_audio_emo.py
@@ -32,7 +32,7 @@ def text_predict(self, texts):
 
         # 初始化tokenizer和模型
         tokenizer = BertTokenizerFast.from_pretrained(tokenizer_path)
-        model = BertForSequenceClassification.from_pretrained(model_path, num_labels=3)  # 修改为你的模型文件路径
+        model = BertForSequenceClassification.from_pretrained(model_path, num_labels=3) 
         model.to(device)
         # 输入：一系列的文本
         # 输出：二维列表，每个子列表包含两个元素，第一个元素是预测结果，第二个元素是预测概率