routers/ner.py

from model.ner import name_entity

class NER(object):
    def __init__(self):
        pass

    def predict(self, req: dict):
        text = req.get("text")
        result = name_entity.predict(text)

        return result

ner = NER()