Update README.md

README.md

@@ -55,7 +55,7 @@ label_mapping = {
     7: "厭惡語調"
 }
 
-def predict_emotion(text, model_path="./fine_tuned_model"):
+def predict_emotion(text, model_path="Johnson8187/chinese-text-emotion-classifier"):
     # 載入模型和分詞器
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)  # 移動模型到設備
@@ -90,6 +90,7 @@ if __name__ == "__main__":
         emotion = predict_emotion(text)
         print(f"文本: {text}")
         print(f"預測情緒: {emotion}\n")
+
 ```
 
 ---