Chris4K commited on
Commit
3f626f1
1 Parent(s): c9b7536

Update sentiment_analysis.py

Browse files
Files changed (1) hide show
  1. sentiment_analysis.py +2 -2
sentiment_analysis.py CHANGED
@@ -27,14 +27,14 @@ class SentimentAnalysisTool:
27
  def get_prediction(model_id):
28
  classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
29
 
30
- def predict(review):
31
  classifier = get_prediction(model_id_7)
32
  prediction = classifier(review)
33
  print(prediction)
34
  return parse_output(prediction)
35
 
36
  def __call__(self, inputs: str):
37
- xxx = predict(str)
38
  return xxx
39
 
40
 
 
27
  def get_prediction(model_id):
28
  classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
29
 
30
+ def predicto(review):
31
  classifier = get_prediction(model_id_7)
32
  prediction = classifier(review)
33
  print(prediction)
34
  return parse_output(prediction)
35
 
36
  def __call__(self, inputs: str):
37
+ xxx = predicto(str)
38
  return xxx
39
 
40