Fix red highlighting

app.py

@@ -172,12 +172,12 @@ def diffusion_chat(question, eot_weight, max_it, sharpness, noise_clipping, use_
         print('Generating output')
         generated_tokens, confidences = generate_diffusion_text(current_tokens, answer_start)
         current_tokens = generated_tokens
-
+        just_noised_indices = []
         decoded_ids = current_tokens[answer_start:]
         decoded_tokens = tokenizer.convert_ids_to_tokens(decoded_ids)
         filtered_tokens = [tok for tok in decoded_tokens if tokenizer.convert_tokens_to_ids(tok) != eot_token_id]
         filtered_prev_tokens = [tok for tok in prev_decoded_tokens if tokenizer.convert_tokens_to_ids(tok) != eot_token_id] if prev_decoded_tokens else []
-        just_noised_indices = []
+        
         if filtered_prev_tokens:
             highlighted = []
             for i, tok in enumerate(decoded_tokens):