Create app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from transformers import pipeline
+
+classifier = pipeline("text-classification", model="WJL110/emotion-classifier")
+
+label_map = {
+    "LABEL_0": "快乐",
+    "LABEL_1": "愤怒",
+    "LABEL_2": "悲伤"
+}
+
+def classify_text(text):
+    result = classifier(text)[0]
+    emotion = label_map.get(result['label'], result['label'])
+    # 返回一个字典，Gradio 会自动将其渲染为分类条形图
+    return {emotion: result['score']}
+
+demo = gr.Interface(
+    fn=classify_text,
+    inputs=gr.Textbox(label="输入文本"),
+    outputs=gr.Label(num_top_classes=1, label="情感预测"),
+    examples=[["今天真是太开心了！"], ["这件事让我很生气。"]],
+    title="情感分析演示"
+)
+
+if __name__ == "__main__":
+    demo.launch()