Fix: correction

app.py

@@ -1018,11 +1018,13 @@ INFERENCE_CHUNK_SIZE = 256
 
 # ========== LABELS (Must match Training Script) ==========
 # Including PASSAGE for the new structuring logic
-LABELS = ["O", "B-QUESTION", "I-QUESTION", "B-OPTION", "I-OPTION", "B-ANSWER", "I-ANSWER", "B-IMAGE", "I-IMAGE", "B-PASSAGE", "I-PASSAGE"]
+# LABELS = ["O", "B-QUESTION", "I-QUESTION", "B-OPTION", "I-OPTION", "B-ANSWER", "I-ANSWER", "B-IMAGE", "I-IMAGE", "B-PASSAGE", "I-PASSAGE"]
+# LABEL2IDX = {l: i for i, l in enumerate(LABELS)}
+# IDX2LABEL = {i: l for i, l in enumerate(LABELS)}
+LABELS = ["O", "B-QUESTION", "I-QUESTION", "B-OPTION", "I-OPTION", "B-ANSWER", "I-ANSWER", "B-IMAGE", "I-IMAGE"]
 LABEL2IDX = {l: i for i, l in enumerate(LABELS)}
 IDX2LABEL = {i: l for i, l in enumerate(LABELS)}
 
-
 # =========================================================
 # 1. Core Classes (Vocab, CharCNNEncoder, MCQTagger)
 # =========================================================