diff --git a/geo-tagger/v1/geoTagger/geoTagger.py b/geo-tagger/v1/geoTagger/geoTagger.py index e4ab9b5..2bd70f1 100755 --- a/geo-tagger/v1/geoTagger/geoTagger.py +++ b/geo-tagger/v1/geoTagger/geoTagger.py @@ -14,12 +14,15 @@ for line in sys.stdin: data = json.loads(line) text=data['value'] - text = Sentence(text) - tagger.predict(text) + sent= text.split(".") + sentences = [Sentence(sent[i]+".") for i in range(len(sent))] + tagger.predict(sentences) geo = [] - for entity in text.get_spans('ner'): - if entity.tag == "LOC": - geo.append(entity.text) + + for sentence in sentences: + for entity in sentence.get_spans('ner'): + if entity.tag == "LOC": + geo.append(entity.text) data['value'] = geo sys.stdout.write(json.dumps(data)) sys.stdout.write('\n')