Johnson8187
/

Chinese-Emotion

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Johnson8187 commited on 10 days ago

Commit

cf1cfc8

•

1 Parent(s): 80330d3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ label_mapping = {
     7: "厭惡語調"
 }
-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)  # 移動模型到設備
@@ -141,7 +141,7 @@ label_mapping = {
     7: "厭惡語調"
 }
-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)  # 移動模型到設備

     7: "厭惡語調"
 }
+def predict_emotion(text, model_path="Johnson8187/Chinese-emotion-classifier"):
     # 載入模型和分詞器
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)  # 移動模型到設備
     7: "厭惡語調"
 }
+def predict_emotion(text, model_path="Johnson8187/Chinese-emotion-classifier"):
     # 載入模型和分詞器
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)  # 移動模型到設備