45d7f12d66addc6235333e70ebefefce96b86150,underthesea/sentiment/general/__init__.py,,sentiment,#Any#,18

Before Change

def sentiment(text):
    global classifier
    if not classifier:
        if os.path.exists(model_path):
            classifier = TextClassifier.load(model_path)
        else:
            logger.error(
                f"Could not load model at {model_path}.\n"
                f"Download model with \"underthesea download {UTSModel.sa_general.value}\".")
            sys.exit(1)
    sentence = Sentence(text)
    classifier.predict(sentence)
    label = sentence.labels[0]
    if label == "1":

After Change

        label = label_map[labels[0]]
        return label
    except Exception:
        return None

Instances