Update app.py

app.py

@@ -107,7 +107,7 @@ def ner(text):
   text = " ".join(rdrsegmenter.word_segment(text))
     
   # Some words in lowercase like "đà nẵng" will get error (due to vncorenlp)
-  text = text.replace("đà ", " đà ")    
+  text = text.replace("_đà ", " đà_").replace("_Đà ", " Đà_")        
     
   output = my_classifier(text)
   for entity in output:
@@ -124,7 +124,10 @@ def ner(text):
 
   slot_tokens = []    
 
-  for word, pred in zip(words, slot_preds):
+  words[0].capitalize()
+  for i, word in enumerate(words):
+      word = word.replace("_", " ")
+      
       if pred == 'O':
           slot_tokens.extend([(word, None), (" ", None)])
       elif pred[0] == 'I':