Update app.py

app.py

@@ -2,19 +2,31 @@ import streamlit as st
 from transformers import pipeline
 
 def main():
-    sentiment_pipeline = pipeline(model="KeviniveK/CustomModel_IMDB")
+    # 1. 建立情感分析 pipeline
+    sentiment_pipeline = pipeline(
+        task="sentiment-analysis",
+        model="distilbert/distilbert-base-uncased-finetuned-sst-2-english"
+    )
 
     st.title("Sentiment Analysis with HuggingFace Spaces")
     st.write("Enter a sentence to analyze its sentiment:")
 
     user_input = st.text_input("")
     if user_input:
-        result = sentiment_pipeline(user_input)
-        sentiment = result[0]["label"]
-        confidence = result[0]["score"]
+        result = sentiment_pipeline(user_input)[0]
+        label = result["label"]       # e.g. "LABEL_0" or "LABEL_1"
+        score = result["score"]
 
-        st.write(f"Sentiment: {sentiment}")
-        st.write(f"Confidence: {confidence:.2f}")
+        # 2. 映射成更友好的文本
+        label_map = {
+            "LABEL_0": "Negative",
+            "LABEL_1": "Positive"
+        }
+        friendly_label = label_map.get(label, label)
+
+        # 3. 输出
+        st.write(f"Sentiment: **{friendly_label}**")
+        st.write(f"Confidence: **{score:.2f}**")
 
 if __name__ == "__main__":
-    main()
+    main()