xujinheng666 commited on
Commit
d253e5c
·
verified ·
1 Parent(s): 9765666

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ rating_pipe = pipeline("sentiment-analysis", model="uer/roberta-base-finetuned-d
54
  def rate_quality(text):
55
  result = rating_pipe(text)[0]
56
  label = result["label"].split("(")[0].strip().lower()
57
- label_map = {"negative": "Poor", "neutral": "Average", "positive": "Good"}
58
  return label_map.get(label, "Unknown")
59
 
60
  # Streamlit UI
 
54
  def rate_quality(text):
55
  result = rating_pipe(text)[0]
56
  label = result["label"].split("(")[0].strip().lower()
57
+ label_map = {"negative": "Poor", "positive": "Good"}
58
  return label_map.get(label, "Unknown")
59
 
60
  # Streamlit UI