Spaces:

xujinheng666
/

CS_Quality_Analysis_FinalProject

Running

xujinheng666 commited on Mar 16

Commit

6589673

verified ·

1 Parent(s): c74678d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ def rate_quality(text):
     result = st.session_state.rating_pipe(text)[0]
     label = result["label"].split("(")[0].strip().lower()
     # label = result["label"]
-    label_map = {"負面": "Poor", "中性": "Average", "正面": "Good"}
     return label_map.get(label, "Unknown")
 def main():

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