Chris4K commited on
Commit
e051ada
1 Parent(s): cc1edab

Update sentiment_analysis.py

Browse files
Files changed (1) hide show
  1. sentiment_analysis.py +1 -1
sentiment_analysis.py CHANGED
@@ -33,7 +33,7 @@ class SentimentAnalysisTool:
33
  classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
34
 
35
  def predicto(review):
36
- classifier = SentimentAnalysisTool.get_prediction(SentimentAnalysisTool.model_id_7)
37
  prediction = classifier(review)
38
  print(prediction)
39
  return SentimentAnalysisTool.parse_output(prediction)
 
33
  classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
34
 
35
  def predicto(review):
36
+ classifier = SentimentAnalysisTool.get_prediction(SentimentAnalysisTool.model_id_3)
37
  prediction = classifier(review)
38
  print(prediction)
39
  return SentimentAnalysisTool.parse_output(prediction)