Update app.py

app.py

@@ -194,12 +194,17 @@ def text_analysis(text):
     for word, pred in zip(words, slot_preds):
         if pred == 'O':
             slot_tokens.extend([(word, None), (" ", None)])
+            
         elif pred[0] == 'I':
             added_tokens = list(slot_tokens[-2]) 
             added_tokens[0] += f' {word}'
             slot_tokens[-2] = tuple(added_tokens)
-        else:
+            
+        elif pred[0] == 'B':
             slot_tokens.extend([(word, pred[2:]), (" ", None)])
+
+        else: # PAD or UNK tags
+            slot_tokens.extend([(word, pred), (" ", None)])
     
     intent_label = intent_label_lst[intent_pred]