modelo menos pesado

app.py

@@ -1,31 +1,31 @@
 import gradio as gr
 from transformers import pipeline
 
+# Modelo mejor que el DeepESP/gpt2-spanish  
 generador = pipeline(
     task="text-generation",
-    model="tiiuae/falcon-7b-instruct",
-    tokenizer="tiiuae/falcon-7b-instruct"
+    model="DeepESP/gpt2-spanish-medium",
+    tokenizer="DeepESP/gpt2-spanish-medium"
 )
 
 def generar_consejo(tema):
-    prompt_contexto = f"Soy un entrenador profesional de fitness y nutrición. Dame un consejo práctico y fácil de seguir sobre {tema}, que pueda aplicar cualquier persona interesada en mejorar su salud."
-    
+    prompt_contexto = f"Consejo útil de salud/fuerza sobre {tema}:"
+
     resultado = generador(
         prompt_contexto,
-        max_length=250,
+        max_length=180,     # un poco más largo para textos más completos
         do_sample=True,
-        temperature=0.7,
-        repetition_penalty=1.2
+        temperature=0.8
     )
-    
+
     return resultado[0]["generated_text"]
 
 demo = gr.Interface(
     fn=generar_consejo,
-    inputs=gr.Textbox(lines=2, placeholder="Ej: mejorar postura, entrenamiento HIIT, hidratación", label="Tema del consejo"),
-    outputs=gr.Textbox(lines=10, label="Consejo generado por la AI"),
-    title="Generador de Consejos Fitness AI",
-    description="Introduce un tema de salud o deporte y la IA generará un consejo práctico y útil.",
+    inputs=gr.Textbox(lines=2, placeholder="Ej: hidratación, fuerza, dormir bien...", label="Tema del consejo"),
+    outputs=gr.Textbox(lines=8, label="Consejo generado"),
+    title="Generador de Consejos Fitness (Mejorado)",
+    description="Introduce un tema de salud o deporte y la IA te dará un consejo práctico."
 )
 
 if __name__ == "__main__":