Newer
Older
web-services / geoEntity-tagger / v1 / geoTagger.py
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')