Fix confidence guided noising call

app.py

@@ -235,7 +235,7 @@ def diffusion_chat(question, max_it, pause_length, sharpness, clustering, noise_
         # --- NOISING STEP ---
         threshold = get_noising_schedule(i, max_it, sharpness=sharpness)
         if use_confidence_noising:
-            noised_answer = confidence_guided_noising(
+            noised_answer, just_noised_indices = confidence_guided_noising(
                 current_tokens, answer_start, confidences, noise_clipping, threshold=threshold, noise_start=noise_start
             )
             just_noised_indices = []