Integrate Flair models about api-inference-community HOT 3 CLOSED

from flair.data import Sentence
from flair.models import SequenceTagger

tagger: SequenceTagger = SequenceTagger.load("dbmdz/flair-historic-ner-onb")

sentence: Sentence = Sentence(input_text)

# Also show scores for recognized NEs
tagger.predict(sentence, all_tag_prob=True, label_name="predicted")

response = []

entities = []

for span in sentence.get_spans("predicted"):
    idx = [token.idx for token in span.tokens]

    current_entity = {
        "entity_group": span.tag,
        "word": span.to_original_text(),
        "start": idx[0] - 1, # Because Flair starts at 1
        "end": idx[-1],
        "score": span.score
    }

    entities.append(current_entity)


response = {
    "entities": entities
}

return jsonify(response)