Update app.py

app.py

@@ -127,18 +127,18 @@ def ner(text):
   for word, pred in zip(words, slot_preds):
       word = word.replace("_", " ")
       if pred == 'O':
-          slot_tokens.extend([(word, None))
+          slot_tokens.extend([(word, None)])
           
       elif pred[0] == 'I': # token with label's prefix' "I-XXX": will be combined their corresponded "B-XXX"
-          added_tokens = list(slot_tokens[-2]) 
+          added_tokens = list(slot_tokens[-1]) 
           added_tokens[0] += f' {word}'
-          slot_tokens[-2] = tuple(added_tokens)
+          slot_tokens[-1] = tuple(added_tokens)
           
       elif pred[0] == 'B': # token with label's prefix "B-XXX" : remove "B-" 
-          slot_tokens.extend([(word, pred[2:]))
+          slot_tokens.extend([(word, pred[2:])])
           
       else: # PAD or UNK tags
-          slot_tokens.extend([(word, pred))
+          slot_tokens.extend([(word, pred)])
   
   intent_label = intent_label_lst[intent_pred]