Logs

app.py

@@ -85,10 +85,10 @@ def align_labels_with_tokens(label, word_ids):
         if word_idx is None:
             aligned_label_ids.append(-100)
         elif word_idx != previous_word_idx:  # Only label the first token of a given word.
-            if label.startswith("B-"):
+            if label[word_idx].startswith("B-"):
                 print(word_idx)
-                print(label)
-                label = label.replace("B-", "I-")
+                print(label[word_idx])
+                label[word_idx] = label[word_idx].replace("B-", "I-")
             aligned_label_ids.append(label[word_idx])
         else:
             aligned_label_ids.append(-100)