#!/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')