Visualize noising

app.py

@@ -256,7 +256,7 @@ def diffusion_chat(question, eot_weight, max_it, pause_length, sharpness, cluste
         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)