Fix pause at start

app.py

@@ -166,8 +166,8 @@ def diffusion_chat(question, eot_weight, max_it, pause_length, sharpness, cluste
     current_tokens, just_noised_indices = noisify_answer(
                 input_ids, answer_start, threshold=1.0, eot_weight=eot_weight, clustering=clustering, noise_start = noise_start,
             )
-    yield f"<b>Iteration 0 (initial noise):</b><br>" + tokenizer.decode(current_tokens, skip_special_tokens=True).replace('\n', '<br>')
-    pause(1)
+        yield f"<b>Iteration 0 (initial noise):</b><br>" + tokenizer.decode(current_tokens[answer_start:], skip_special_tokens=True).replace('\n', '<br>')
+    time.sleep(pause_length)
     last_tokens = []
     prev_decoded_tokens = []