Update app.py

app.py

@@ -569,13 +569,11 @@ def live_denoise(src, state_ids, steps, snap_every, seed, mode, temperature, exc
 # ===============================================
 # UI (single mode)
 # ===============================================
-with gr.Blocks(title="CNA — Interactive Denoising") as demo:
+with gr.Blocks(title="Self Organising Text Demo") as demo:
     gr.Markdown(
         """
-        # CNA — Interactive Denoising (Strategy 1)
-        - **Weights source**: `.pt` weights-only (e.g., `weights_latest.pt`), a folder of shards, or a **Hub repo id**.
-        - Update rule per step: **argmax** or **sample** (temperature + exclude current).
-        - Tools: Random init, **drag to select** in the text box → *Noise Selection*, manual indices, prepend/append noise, live denoise.
+        # Self Organising Text Demo
+        Watch text self organise using only local attention.
         """
     )
 
@@ -585,7 +583,7 @@ with gr.Blocks(title="CNA — Interactive Denoising") as demo:
 
     with gr.Row():
         src = gr.Textbox(value=default_source, label="Weights (file / folder / HF repo id)")
-        seqlen = gr.Slider(10, 512, value=100, step=1, label="Sequence length (S)")
+        seqlen = gr.Slider(10, 512, value=50, step=1, label="Sequence length (S)")
         seed = gr.Slider(0, 10000, value=0, step=1, label="Seed")
 
     ids_state = gr.State(value=None)
@@ -597,16 +595,20 @@ with gr.Blocks(title="CNA — Interactive Denoising") as demo:
     gr.Markdown("### Initialize & Denoise")
     with gr.Row():
         btn_random = gr.Button("Initialize Random")
-        steps = gr.Slider(1, 2000, value=200, step=1, label="Denoise steps (N)")
-        snap_every = gr.Slider(1, 100, value=5, step=1, label="Update every K steps")
+        steps = gr.Slider(1, 2000, value=100, step=1, label="Denoise steps (N)")   # default 100
+        snap_every = gr.Slider(1, 100, value=1, step=1, label="Update every K steps")  # default 1
     with gr.Row():
-        update_mode = gr.Radio(choices=["argmax", "sample"], value="argmax", label="Update rule")
+        update_mode = gr.Radio(
+            choices=["argmax", "sample"],
+            value="sample",   # default to sampling
+            label="Update rule"
+        )
         temperature = gr.Slider(minimum=0.0, maximum=5.0, value=1.0, step=0.05, label="Temperature (sampling)")
         exclude_current = gr.Checkbox(value=True, label="Exclude current token when sampling")
     with gr.Row():
         btn_step_once = gr.Button("Step Once")
         btn_live = gr.Button("Denoise Live (streaming)")
-
+    
     gr.Markdown("### Noise by Indices")
     with gr.Row():
         indices_csv = gr.Textbox(
@@ -617,6 +619,7 @@ with gr.Blocks(title="CNA — Interactive Denoising") as demo:
         add_left = gr.Number(value=0, precision=0, label="Noise tokens to add at START")
         add_right = gr.Number(value=0, precision=0, label="Noise tokens to add at END")
         btn_apply_noise = gr.Button("Apply Noise")
+    
 
 
     # --- Wiring ---