Upload app.py with huggingface_hub

app.py

@@ -145,23 +145,116 @@ demo = gr.ChatInterface(
     fn=chat,
     title="🧠 Titans + MIRAS: A Brain That Changes Itself While Thinking",
     description="""
-    This chatbot uses **test-time learning** - it updates its memory with every message!
+    ## 🚀 The Revolutionary Difference
     
-    **How it works:**
-    1. Your message is processed through distilgpt2
-    2. Memory predicts what it should remember
-    3. Prediction error (loss) indicates surprise
-    4. Higher surprise → stronger memory formation
-    5. Memory weights update via gradient descent
-    6. Response generated and memory saved to disk
+    **Standard LLMs (ChatGPT, Claude, etc.)**: Think → Predict → **Forget**  
+    **Titans + MIRAS**: Think → Predict → **Update** → **Remember** → Think Differently
     
-    **Watch the stats below each response to see the brain learning!**
+    ---
+    
+    ### 💡 What Makes This Different?
+    
+    | Feature | ChatGPT/Claude/GPT-4 | This Demo (Titans+MIRAS) |
+    |---------|---------------------|--------------------------|
+    | **Weights during chat** | 🔒 Frozen forever | ✅ Update with every message |
+    | **Learning** | ❌ Simulated (in-context only) | ✅ Real (gradient descent) |
+    | **Memory** | 📝 Token context only | 🧠 Neural parameters |
+    | **Persistence** | ❌ Forgets when context ends | ✅ Saves to disk |
+    | **Adaptation** | 🎭 Acts like it learned | 🔬 Actually learns |
+    
+    ---
+    
+    ### 🎯 What You're Witnessing
+    
+    **This is NOT a better chatbot** - it's a **learning demonstrator**.
+    
+    1. **The text responses are random** - that's expected! We're using a small, frozen model (distilgpt2)
+    2. **The MAGIC is in the numbers below** - watch the "Loss" decrease when you repeat inputs!
+    3. **Every message physically changes the brain** - the memory weights update via gradient descent
+    4. **Refresh the page** - the update count continues (memory persists!)
+    
+    ---
+    
+    ### 🧪 How It Works (The Technical Truth)
+    
+    ```
+    Your Message
+        ↓
+    [distilgpt2: FROZEN] ← Not learning, just generating
+        ↓
+    Hidden States (768-dim)
+        ↓
+    [Projections] → Memory Space (256-dim)
+        ↓
+    [MIRAS Memory: LEARNING!] ← This is what updates!
+        ↓
+    Loss = How surprised the memory is
+        ↓
+    Gradient Descent → Memory weights change
+        ↓
+    Saved to disk → Persists forever
+    ```
+    
+    **Key Insight**: We're training the **memory**, not the text generator!
+    
+    ---
+    
+    ### 🔬 The Science: Why This Matters
+    
+    **Standard LLMs**:
+    - Weights frozen after training (costs millions)
+    - "Learning" is just pattern matching in context
+    - Forget everything when context ends
+    - Same model for everyone
+    
+    **Titans + MIRAS**:
+    - Weights update during inference (free!)
+    - Real optimization via gradient descent
+    - Memory persists across sessions
+    - Personalizes to each user
+    
+    **This is test-time learning** - the future of adaptive AI.
+    
+    ---
+    
+    ### 📊 What the Stats Mean
+    
+    - **Loss**: How surprised the memory is (lower = more familiar)
+    - **Retention**: Learning rate multiplier (2.0x = very surprising, 0.5x = familiar)
+    - **Updates**: Total number of memory updates (persists across sessions!)
+    - **Avg Loss**: Overall learning progress
+    
+    ---
+    
+    ### 🎮 Try This Experiment
+    
+    1. **Send "hello world" 5 times** → Watch loss decrease!
+    2. **Send something completely different** → Loss spikes!
+    3. **Refresh the page and send another message** → Update count continues!
+    
+    **That decreasing loss is proof the neural weights are changing!**
+    
+    ---
+    
+    ### 🌟 The Bottom Line
+    
+    **ChatGPT**: A frozen calculator that *simulates* adaptation  
+    **This Demo**: A living system that *performs* adaptation
+    
+    You're not chatting with a model.  
+    **You're watching a brain rewire itself in real-time.** 🧠⚡
+    
+    ---
+    
+    *Built with Titans (test-time training) + MIRAS (associative memory)*  
+    *Papers: [Titans](https://arxiv.org/abs/2501.00663) | [MIRAS](https://arxiv.org/abs/2504.13173)*
     """,
     examples=[
-        "Hello! What can you do?",
+        "hello world",
+        "hello world",  # Repeat to show learning!
         "Tell me about test-time learning",
         "What is 2+2?",
-        "Repeat this exact phrase: The quick brown fox",
+        "my name is [your name]",
     ],
     cache_examples=False,
     theme="soft",