Chris4K commited on
Commit
5cb8034
1 Parent(s): ab6a84b

Update sentiment_analysis.py

Browse files
Files changed (1) hide show
  1. sentiment_analysis.py +3 -3
sentiment_analysis.py CHANGED
@@ -5,7 +5,7 @@ from transformers import Tool
5
 
6
  class SentimentAnalysisTool(Tool):
7
  name = "sentiment_analysis"
8
- description = "This tool analyses the sentiment of a given text input."
9
 
10
  inputs = ["text"] # Adding an empty list for inputs
11
  outputs = ["json"]
@@ -18,8 +18,8 @@ class SentimentAnalysisTool(Tool):
18
  model_id_6 = "sbcBI/sentiment_analysis_model"
19
  model_id_7 = "models/oliverguhr/german-sentiment-bert"
20
 
21
- def __call__(self, inputs: str):
22
- return self.predicto(inputs)
23
 
24
  def parse_output(self, output_json):
25
  list_pred = []
 
5
 
6
  class SentimentAnalysisTool(Tool):
7
  name = "sentiment_analysis"
8
+ description = "This tool analyses the sentiment of a given text."
9
 
10
  inputs = ["text"] # Adding an empty list for inputs
11
  outputs = ["json"]
 
18
  model_id_6 = "sbcBI/sentiment_analysis_model"
19
  model_id_7 = "models/oliverguhr/german-sentiment-bert"
20
 
21
+ def __call__(self, text: str):
22
+ return self.predicto(text)
23
 
24
  def parse_output(self, output_json):
25
  list_pred = []