Spaces:

xujinheng666
/

CS_Quality_Analysis_FinalProject

Running

xujinheng666 commited on Mar 16

Commit

5dca2b3

verified ·

1 Parent(s): 3f91072

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def translate_text(text):
 def rate_quality(text):
     result = st.session_state.rating_pipe(text)[0]
-    label = result["label"].lower()
     label_map = {"negative": "Poor", "neutral": "Average", "positive": "Good"}
     return label_map.get(label, "Unknown")

 def rate_quality(text):
     result = st.session_state.rating_pipe(text)[0]
+    label = result["label"].split("(")[0].strip().lower()
     label_map = {"negative": "Poor", "neutral": "Average", "positive": "Good"}
     return label_map.get(label, "Unknown")