textclassifier.py

class TextClassifier:

    def __init__(self, model) -> None:
        self.text_classifier = model

    def classify(self,text):

        data = self.text_classifier(text)[0]
        print("response from hugging \n", data)

        return data