Upload agent.py

agent.py

@@ -0,0 +1,15 @@
+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)
+
+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