Spaces:

xujinheng666
/

CS_Quality_Analysis_FinalProject

Running

xujinheng666 commited on Mar 18

Commit

95e8e86

verified ·

1 Parent(s): d253e5c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def translate(text):
     return " ".join(translations)
 # Load sentiment analysis model
-rating_pipe = pipeline("sentiment-analysis", model="uer/roberta-base-finetuned-dianping-chinese")
 def rate_quality(text):
     result = rating_pipe(text)[0]

     return " ".join(translations)
 # Load sentiment analysis model
+rating_pipe = pipeline("text-classification", model="Leo0129/CustomModel_dianping-chinese")
 def rate_quality(text):
     result = rating_pipe(text)[0]