Update app.py

app.py

@@ -3,7 +3,7 @@ import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-model_name = "rubenroy/Zurich-14B-GCv2-1m"
+model_name = "rubenroy/Zurich-14B-GCv2-5m"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,
@@ -112,10 +112,10 @@ TITLE_HTML = """
         <div style="display: flex; align-items: center; justify-content: center; gap: 1rem;">
             <h1 style="font-size: 2.5rem; font-weight: 800; margin: 0; background: linear-gradient(135deg, #60a5fa 0%, #93c5fd 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">Zurich</h1>
             <div style="width: 2px; height: 2.5rem; background: linear-gradient(180deg, #3b82f6 0%, #60a5fa 100%);"></div>
-            <p style="font-size: 1.25rem; color: #94a3b8; margin: 0;">GammaCorpus v2-1m</p>
+            <p style="font-size: 1.25rem; color: #94a3b8; margin: 0;">GammaCorpus v2-5m</p>
         </div>
         <div class="info-section">
-            <span>Fine-tuned from <a href="https://huggingface.co/Qwen/Qwen2.5-14B-Instruct" class="info-link">Qwen 2.5 14B Instruct</a> | Model: <a href="https://huggingface.co/rubenroy/Zurich-14B-GCv2-1m" class="info-link">Zurich-14B-GCv2-1m</a> | Training Dataset: <a href="https://huggingface.co/datasets/rubenroy/GammaCorpus-v2-1m" class="info-link">GammaCorpus v2 1m</a></span>
+            <span>Fine-tuned from <a href="https://huggingface.co/Qwen/Qwen2.5-14B-Instruct" class="info-link">Qwen 2.5 14B Instruct</a> | Model: <a href="https://huggingface.co/rubenroy/Zurich-14B-GCv2-5m" class="info-link">Zurich-14B-GCv2-5m</a> | Training Dataset: <a href="https://huggingface.co/datasets/rubenroy/GammaCorpus-v2-5m" class="info-link">GammaCorpus v2 5m</a></span>
         </div>
     </div>