Update app.py

app.py

@@ -4,36 +4,36 @@ import torch
 import os
 from huggingface_hub import login
 
-# Login con token seguro
+# Login Hugging Face
 hf_token = os.environ["HF_TOKEN"]
 login(token=hf_token)
 
-# Usa GPU si hay
+# GPU si hay
 device = 0 if torch.cuda.is_available() else -1
 
-# Modelo de tipo chat liviano
+# Cargar TinyLlama chat
 pipe = pipeline(
     "text-generation",
-    model="mistralai/TinyMistral-248M-Chat-v1",
+    model="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
     device=device
 )
 
-# Formato chat para prompts
+# Formato tipo chat
 def responder(prompt):
-    formatted_prompt = f"[INST] {prompt} [/INST]"
-    respuesta = pipe(
-        formatted_prompt,
-        max_new_tokens=80,
+    prompt_chat = f"<|system|>Eres un asistente útil.<|user|>{prompt}<|assistant|>"
+    output = pipe(
+        prompt_chat,
+        max_new_tokens=100,
         do_sample=True,
         temperature=0.7,
-        top_k=50,
         top_p=0.9
-    )[0]["generated_text"]
-    return respuesta.replace(formatted_prompt, "").strip()
+    )[0]['generated_text']
+    respuesta = output.replace(prompt_chat, "").strip()
+    return respuesta
 
-# Interfaz Gradio
+# Interfaz
 with gr.Blocks() as demo:
-    gr.Markdown("## ⚡ AmInside 1.0 – Versión Chat Ligera")
+    gr.Markdown("## 🤖 AmInside 1.0 – Asistente rápido y conversacional")
     entrada = gr.Textbox(label="Escribe tu mensaje")
     salida = gr.Textbox(label="Respuesta")
     entrada.submit(fn=responder, inputs=entrada, outputs=salida)