Add newlines to output

app.py

@@ -173,7 +173,7 @@ def diffusion_chat(question, eot_weight, max_it, sharpness, noise_clipping, use_
             highlighted = [tokenizer.convert_tokens_to_string([tok]) for tok in filtered_tokens]
 
         prev_decoded_tokens = decoded_tokens
-        yield f"<b>Iteration {i+1}/{max_it} (running):</b><br>" + "".join(highlighted)
+        yield f"<b>Iteration {i+1}/{max_it} (running):</b><br>" + "".join(highlighted).replace('\n', '<br>')
 
         last_tokens.append(generated_tokens)
         if len(last_tokens) > 3:
@@ -194,7 +194,7 @@ def diffusion_chat(question, eot_weight, max_it, sharpness, noise_clipping, use_
     final_tokens = [tok for tok in final_tokens if tokenizer.convert_tokens_to_ids(tok) != eot_token_id]
     final_output = tokenizer.convert_tokens_to_string(final_tokens)
     print(final_output)
-    yield f"<b>Final Output (after {i+1} iterations):</b><br>" + final_output
+    yield f"<b>Final Output (after {i+1} iterations):</b><br>" + final_output.replace('\n', '<br>')
 
 # --- Gradio Interface ---