Update interface

app.py

@@ -78,16 +78,17 @@ def highlight_tokens(token_ids, answer_start, changed_indices, color):
             highlighted.append(tok_str)
     return "".join(highlighted)
 
-def diffusion_chat(question, noising, max_it, pause_length):
+def diffusion_chat(question, noising, enable_pause, max_it):
     
     sharpness = 3.0
     noise_start = 0.5
     top_p = 1.0
     top_k = 10
     clustering = False
+    pause_length = 1.0 if enable_pause else 0.0
 
     if question.strip() == "":
-        question = "What do you know about the city of Amsterdam?"
+        question = "What do you know about Amsterdam?"
 
     prompt = format_chat_prompt(question)
     input_ids = tokenizer.encode(prompt, add_special_tokens=False)
@@ -200,15 +201,20 @@ assistant_marker_ids = tokenizer.encode("<|start_header_id|>assistant<|end_heade
 demo = gr.Interface(
     fn=diffusion_chat,
     inputs=[
-        gr.Textbox(label="User Question", lines=2, placeholder="What do you know about the city of Amsterdam?"),
-        gr.Checkbox(label="Enable noising", value=True, info="If disabled, the model will not apply any intermediate noise."),
-        gr.Slider(1, 512, value=64, step=1, label="Increase the maximum number of iterations to run."),
-        gr.Slider(0, 5, value=0, step=0.01, label="Increase the pause between iterations to visualize the process.")
+        gr.Textbox(
+            label="User Question",
+            lines=2,
+            placeholder="What do you know about Amsterdam?",
+        ),
+        gr.Checkbox(label="Enable intermediate noising", value=True),
+        gr.Checkbox(label="Pause between iterations", value=False),
+        gr.Slider(1, 512, value=64, step=1, label="Increase the maximum number of iterations."),
     ],
-    outputs=[gr.HTML(label="Diffusion Output")],
-    title="Diffusion Language Model Chat",
-    theme="default",
-    description="This interface runs a diffusion-based language model to generate answers progressively."
+    outputs=gr.HTML(label="Diffusion Output"),
+    title="LAD Chat",
+    description="This interface runs a diffusion-based language model to generate answers progressively.",
+    allow_flagging="never",
+    live=False  # ensures the Stop button appears properly
 )
 
 demo.launch(share=True, allowed_paths=["."], ssr_mode=False)