Update sentiment_analysis.py

sentiment_analysis.py

@@ -1,8 +1,9 @@
 import requests
 import gradio as gr
 from transformers import pipeline
+from transformers import Tool
 
-class SentimentAnalysisTool:
+class SentimentAnalysisTool(Tool):
     name = "sentiment_analysis"
     description = "This tool analyses the sentiment of a given text input."
 
@@ -11,7 +12,7 @@ class SentimentAnalysisTool:
     outputs = ["json"]
     
     def __call__(self, inputs: str): 
-        return SentimentAnalysisTool.predicto(str)
+        return predicto(str)
         
     model_id_1 = "nlptown/bert-base-multilingual-uncased-sentiment"
     model_id_2 = "microsoft/deberta-xlarge-mnli"
@@ -33,10 +34,10 @@ class SentimentAnalysisTool:
         classifier = pipeline("text-classification", model=model_id, return_all_scores=True)
                  
     def predicto(review):
-            classifier = SentimentAnalysisTool.get_prediction(SentimentAnalysisTool.model_id_3)
+            classifier = get_prediction(model_id_3)
             prediction = classifier(review)
             print(prediction)
-            return SentimentAnalysisTool.parse_output(prediction)
+            return parse_output(prediction)