Update agent.py

agent.py

@@ -1,15 +1,16 @@
 from transformers import pipeline
 
-# Load fine-tuned translation model
-translator = pipeline("text-generation", model="JenniferHJF/qwen1.5-emoji-finetuned", max_new_tokens=20, device=0)
-
-# Load hate speech classifier model
-classifier = pipeline("text-classification", model="unitary/toxic-bert", top_k=None, device=0)
+# Load both pipelines
+translator = pipeline("text-generation", model="JenniferHJF/qwen1.5-emoji-finetuned", max_new_tokens=20)
+classifier = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-offensive")
 
 def classify_emoji_text(text):
-    translated = translator(f"输入：{text}\n输出：")[0]["generated_text"]
-    translated_text = translated.strip().split("\n")[-1]
-    result = classifier(translated_text)[0]
-    label = result["label"]
-    score = result["score"]
-    return translated_text, label, score
+    translated_output = translator(text)[0]["generated_text"]
+    translated = translated_output.strip().split("\n")[-1]  # 提取最后一行作为翻译结果
+
+    result = classifier(translated)
+    first = result[0]  # ✅ 修复错误：取列表第一个元素
+    label = first["label"]
+    score = first["score"]
+
+    return translated, label, score