Update app.py

app.py

@@ -1,18 +1,31 @@
 from transformers import pipeline
 import gradio as gr
 
-classifier = pipeline("sentiment-analysis")
+# 載入 Hugging Face 的中文三分類情緒模型
+classifier = pipeline("text-classification", model="uer/roberta-base-finetuned-dianping-chinese", tokenizer="uer/roberta-base-finetuned-dianping-chinese")
 
-def analyze(text):
+# 分析函式
+def analyze_sentiment(text):
     result = classifier(text)[0]
     label = result["label"]
-    score = round(result["score"], 3)
-    return f"情緒：{label}\n信心分數：{score}"
+    score = round(result["score"], 4)
 
-demo = gr.Interface(fn=analyze,
-                    inputs=gr.Textbox(label="輸入文字"),
-                    outputs="text",
-                    title="情緒分析器",
-                    description="輸入一段文字，判斷其情緒為 Positive 或 Negative")
+    if label == "0":
+        sentiment = "負向情緒"
+    elif label == "1":
+        sentiment = "中立情緒"
+    elif label == "2":
+        sentiment = "正向情緒"
+    else:
+        sentiment = "未知"
 
-demo.launch()
+    return f"判斷結果：{sentiment}\n信心分數：{score}"
+
+# Gradio 介面
+gr.Interface(
+    fn=analyze_sentiment,
+    inputs=gr.Textbox(lines=4, placeholder="請輸入中文內容"),
+    outputs="text",
+    title="中文情緒分析系統（正向／中立／負向）",
+    description="輸入一段中文，模型會自動判斷情緒傾向"
+).launch()