Newer
Older
web-services / geo-tagger / v1 / geoTagger / geoTagger.py
@Lucas Anki Lucas Anki on 3 Mar 2022 597 bytes fix
#!/opt/bitnami/python/bin/python3.7

import sys
import json

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

import logging
logging.getLogger('flair').handlers[0].stream = sys.stderr

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')