Disfluency-large

Sleeping

DD0101 commited on May 3, 2023

Commit

8a2ba57

1 Parent(s): e7a0f80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,9 +125,8 @@ def ner(text):
   slot_tokens = []
   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':

   slot_tokens = []
   words[0].capitalize()
+  for word, pred in zip(words, slot_preds):
       word = word.replace("_", " ")
       if pred == 'O':
           slot_tokens.extend([(word, None), (" ", None)])
       elif pred[0] == 'I':