Update app.py

app.py

@@ -1,4 +1,4 @@
-# HF Space · WFGY demo (all-English, no comments in other languages)
+# HF Space · WFGY variance gate demo (Gradio 4.31+)
 
 import io
 import numpy as np
@@ -8,27 +8,27 @@ from wfgy_sdk import get_engine
 from wfgy_sdk.evaluator import compare_logits, plot_histogram
 
 MODEL_ID = "sshleifer/tiny-gpt2"
-
 tok = AutoTokenizer.from_pretrained(MODEL_ID)
 mdl = AutoModelForCausalLM.from_pretrained(MODEL_ID)
 eng = get_engine()
 
+
 def run(prompt: str):
     prompt = prompt.strip()
     if not prompt:
-        return "", "", "No prompt — nothing to show", None
+        return "", "", "no prompt – nothing to show", None
 
     ids = tok(prompt, return_tensors="pt").input_ids
     logits_raw = mdl(ids).logits[0, -1].detach().cpu().numpy()
 
-    # toy fingerprints just for the demo
+    # toy fingerprints
     G = np.random.randn(256).astype(np.float32)
     I = G + np.random.normal(scale=0.05, size=256).astype(np.float32)
 
     logits_mod = eng.run(I, G, logits_raw)
     m = compare_logits(logits_raw, logits_mod)
 
-    headline = f"▼ var {m['var_drop']*100:.1f} % | KL {m['kl']:.3f}"
+    headline = f"▼ var {m['var_drop']*100:4.1f} % | KL {m['kl']:.3f}"
 
     fig = plot_histogram(logits_raw, logits_mod)
     buf = io.BytesIO()
@@ -40,10 +40,10 @@ def run(prompt: str):
     return raw_txt, mod_txt, headline, buf
 
 
-with gr.Blocks(title="WFGY Variance Gate") as demo:
+with gr.Blocks(title="WFGY variance gate") as demo:
     gr.Markdown(
-        "# 🧠 WFGY simulation demo\n"
-        "Type any prompt – watch variance shrink in real time."
+        "# 🧠 WFGY simulation demo  \n"
+        "Type any prompt and watch the logit variance collapse in real time."
     )
 
     prompt = gr.Textbox(label="Prompt", value="Explain Schrödinger's cat")
@@ -60,9 +60,10 @@ with gr.Blocks(title="WFGY Variance Gate") as demo:
 
     gr.Markdown(
         "---\n"
-        "### ⭐ Help unlock **WFGY 2.0**\n"
-        "10 000 stars on GitHub by **2025-08-01** → next-gen release."
+        "### ⭐ Help unlock **WFGY 2.0**  \n"
+        "10 000 GitHub stars by **2025-08-01** → next-gen release."
     )
 
 if __name__ == "__main__":
-    demo.queue(concurrency_count=2).launch()
+    # Gradio ≥4.31: queue() has no arg; use default queue size (=2)
+    demo.queue().launch()