Update Bert_predict.py

Bert_predict.py

@@ -9,11 +9,15 @@ import requests
 import time
 from typing import List, Optional, Dict, Any
 
+from transformers import AutoTokenizer
+### huawei-noah/TinyBERT_General_4L_312D huawei-noah/TinyBERT_General_6L_768D hfl/rbt3 bert-base-chinese
+Bert_model_name = "hfl/rbt3"
+
 class BertPredictor:
     """
     用於加載 BERT 模型、獲取新聞並對其進行股市影響預測的類別。
     """
-    def __init__(self, tokenizer_name: str = 'hfl/rbt3', max_news_per_keyword: int = 5):
+    def __init__(self, tokenizer_name: str = Bert_model_name, max_news_per_keyword: int = 5):
         """
         初始化預測器，載入分詞器、預訓練模型並獲取新聞。
 
@@ -43,7 +47,8 @@ class BertPredictor:
 
         # --- 模型相關設置 ---
         self.text_max_length = 256
-        self.tokenizer = BertTokenizer.from_pretrained(tokenizer_name)
+        self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
+        #self.tokenizer = BertTokenizer.from_pretrained(tokenizer_name)
         
         # 載入最佳模型
         print("正在加載模型...")