Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -194,12 +194,17 @@ def text_analysis(text):
|
|
194 |
for word, pred in zip(words, slot_preds):
|
195 |
if pred == 'O':
|
196 |
slot_tokens.extend([(word, None), (" ", None)])
|
|
|
197 |
elif pred[0] == 'I':
|
198 |
added_tokens = list(slot_tokens[-2])
|
199 |
added_tokens[0] += f' {word}'
|
200 |
slot_tokens[-2] = tuple(added_tokens)
|
201 |
-
|
|
|
202 |
slot_tokens.extend([(word, pred[2:]), (" ", None)])
|
|
|
|
|
|
|
203 |
|
204 |
intent_label = intent_label_lst[intent_pred]
|
205 |
|
|
|
194 |
for word, pred in zip(words, slot_preds):
|
195 |
if pred == 'O':
|
196 |
slot_tokens.extend([(word, None), (" ", None)])
|
197 |
+
|
198 |
elif pred[0] == 'I':
|
199 |
added_tokens = list(slot_tokens[-2])
|
200 |
added_tokens[0] += f' {word}'
|
201 |
slot_tokens[-2] = tuple(added_tokens)
|
202 |
+
|
203 |
+
elif pred[0] == 'B':
|
204 |
slot_tokens.extend([(word, pred[2:]), (" ", None)])
|
205 |
+
|
206 |
+
else: # PAD or UNK tags
|
207 |
+
slot_tokens.extend([(word, pred), (" ", None)])
|
208 |
|
209 |
intent_label = intent_label_lst[intent_pred]
|
210 |
|