Chris4K commited on
Commit
3875537
1 Parent(s): 1843ae2

Update sentiment_analysis.py

Browse files
Files changed (1) hide show
  1. sentiment_analysis.py +3 -5
sentiment_analysis.py CHANGED
@@ -15,11 +15,6 @@ class SentimentAnalysisTool:
15
  model_id_5 = "juliensimon/reviews-sentiment-analysis"
16
  model_id_6 = "sbcBI/sentiment_analysis_model"
17
  model_id_7 = "models/oliverguhr/german-sentiment-bert"
18
-
19
- def __call__(self, inputs: str):
20
- return predict(str)
21
-
22
-
23
 
24
  def parse_output(output_json):
25
  list_pred=[]
@@ -37,6 +32,9 @@ class SentimentAnalysisTool:
37
  prediction = classifier(review)
38
  print(prediction)
39
  return parse_output(prediction)
 
 
 
40
 
41
 
42
 
 
15
  model_id_5 = "juliensimon/reviews-sentiment-analysis"
16
  model_id_6 = "sbcBI/sentiment_analysis_model"
17
  model_id_7 = "models/oliverguhr/german-sentiment-bert"
 
 
 
 
 
18
 
19
  def parse_output(output_json):
20
  list_pred=[]
 
32
  prediction = classifier(review)
33
  print(prediction)
34
  return parse_output(prediction)
35
+
36
+ def __call__(self, inputs: str):
37
+ return predict(str)
38
 
39
 
40