Johnson8187
/

Chinese-Emotion

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Johnson8187 commited on about 13 hours ago

Commit

6b75089

•

1 Parent(s): 8629a02

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -75,7 +75,7 @@ label_mapping = {
     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)  # 移動模型到設備
@@ -143,7 +143,7 @@ label_mapping = {
     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"):
     # 載入模型和分詞器
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)  # 移動模型到設備
     7: "厭惡語調"
 }
+def predict_emotion(text, model_path="Johnson8187/Chinese-Emotion"):
     # 載入模型和分詞器
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)  # 移動模型到設備