Change red highlighting for debugging

app.py

@@ -269,12 +269,12 @@ def diffusion_chat(question, eot_weight, mask_weight, max_it, pause_length, shar
         current_tokens = ori_input_tokens[:answer_start] + noised_answer[answer_start:]
 
         # --- RED HIGHLIGHT ---
-        decoded_tokens = tokenizer.convert_ids_to_tokens(previous_tokens[answer_start:])
+        decoded_tokens = tokenizer.convert_ids_to_tokens(current_tokens[answer_start:])
         highlighted = []
         for j, tok in enumerate(decoded_tokens):
             tok_id = tokenizer.convert_tokens_to_ids(tok)
-            if tok_id == eot_token_id:
-                continue
+            # if tok_id == eot_token_id:
+            #     continue
             token_str = tokenizer.convert_tokens_to_string([tok])
             abs_idx = answer_start + j
             if abs_idx in just_noised_indices: