ValadisCERTH commited on
Commit
a7fdc57
·
1 Parent(s): 33a5b65

Update helper.py

Browse files
Files changed (1) hide show
  1. helper.py +1 -3
helper.py CHANGED
@@ -256,9 +256,6 @@ def identify_locations(sentence):
256
  locations = []
257
 
258
  try:
259
-
260
- # this is because there were cases were a city followed by comma was not understood by the system
261
- sentence = sentence.replace(",", " $ ")
262
 
263
  # ner
264
  locations.append(identify_loc_ner(sentence))
@@ -326,6 +323,7 @@ def identify_locations(sentence):
326
 
327
  # if a city does not exist
328
  else:
 
329
  # we only accept for one country
330
  if len(locations_dict['country']) == 1:
331
  locations_dict['country'][0] = locations_dict['country'][0].capitalize()
 
256
  locations = []
257
 
258
  try:
 
 
 
259
 
260
  # ner
261
  locations.append(identify_loc_ner(sentence))
 
323
 
324
  # if a city does not exist
325
  else:
326
+
327
  # we only accept for one country
328
  if len(locations_dict['country']) == 1:
329
  locations_dict['country'][0] = locations_dict['country'][0].capitalize()