Update app.py

app.py

@@ -54,7 +54,7 @@ rating_pipe = pipeline("text-classification", model="Leo0129/CustomModel_dianpin
 def rate_quality(text):
     result = rating_pipe(text)[0]
     label = result["label"].split("(")[0].strip().lower()
-    label_map = {"negative": "Poor", "positive": "Good"}
+    label_map = {"LABEL_0": "Poor", "LABEL_1": "Neutral", "LABEL_2": "Good"}
     return label_map.get(label, "Unknown")
 
 # Streamlit UI