Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 # =========================
 # CONFIG
 # =========================
-MODEL_ID = "CromIA/MicroLM-1M"
+MODEL_ID = "CromIA/MicroLM2-1M"
 
 # =========================
 # LOAD MODEL
@@ -53,8 +53,8 @@ demo = gr.Interface(
         gr.Slider(0.5, 1.0, value=0.95, label="Top-p"),
     ],
     outputs=gr.Textbox(label="Output"),
-    title="MicroLM-1M",
-    description="Modelo de linguagem leve (~1M parâmetros) treinado em 500M tokens."
+    title="MicroLM2-1M",
+    description="Modelo de linguagem leve (~1M parâmetros) treinado em 4.5B tokens."
 )
 
 # =========================