sentiment-tool

Sleeping

Chris4K commited on Dec 18, 2023

Commit

e051ada

•

1 Parent(s): cc1edab

Update sentiment_analysis.py

Files changed (1) hide show

sentiment_analysis.py CHANGED Viewed

@@ -33,7 +33,7 @@ class SentimentAnalysisTool:
         classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
     def predicto(review):
-            classifier = SentimentAnalysisTool.get_prediction(SentimentAnalysisTool.model_id_7)
             prediction = classifier(review)
             print(prediction)
             return SentimentAnalysisTool.parse_output(prediction)

         classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
     def predicto(review):
+            classifier = SentimentAnalysisTool.get_prediction(SentimentAnalysisTool.model_id_3)
             prediction = classifier(review)
             print(prediction)
             return SentimentAnalysisTool.parse_output(prediction)