import sys import json from flair.models import SequenceTagger from flair.data import Sentence tagger = SequenceTagger.load("flair/ner-english") for line in sys.stdin: data = json.loads(line) text=data['value'] text = Sentence(text) tagger.predict(text) geo = [] for entity in text.get_spans('ner'): if entity.tag == "LOC": geo.append(entity.text) data['value'] = geo sys.stdout.write(json.dumps(data)) sys.stdout.write('\n')