nuevo modelo

app.py

@@ -1,47 +1,32 @@
 import gradio as gr
 from transformers import pipeline
 
-# 1. Inicialización del pipeline
-# Cambiamos a 'text-generation' usando un modelo GPT-2 entrenado en español.
-# Este modelo es ideal para completar frases o generar ideas para blogs/redes sociales de fitness.
 generador = pipeline(
     task="text-generation",
-    model="DeepESP/gpt2-spanish",
-    tokenizer="DeepESP/gpt2-spanish"
+    model="tiiuae/falcon-7b-instruct",
+    tokenizer="tiiuae/falcon-7b-instruct"
 )
 
 def generar_consejo(tema):
-    # Añadimos un pequeño "prompt" para guiar al modelo hacia el contexto de salud
-    prompt_contexto = f"Consejo saludable sobre {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."
     
     resultado = generador(
         prompt_contexto,
-        max_length=150,     # Longitud máxima del texto generado
-        do_sample=True,     # Permite variedad en la respuesta (creatividad)
-        temperature=0.7,    # Controla la creatividad (0.7 es equilibrado)
-        top_k=50            # Filtra las palabras más probables para evitar incoherencias
+        max_length=250,
+        do_sample=True,
+        temperature=0.7,
+        repetition_penalty=1.2
     )
     
-    print(resultado) # Para depuración en consola
     return resultado[0]["generated_text"]
 
-# 2. Interfaz Gráfica
 demo = gr.Interface(
     fn=generar_consejo,
-    inputs=gr.Textbox(
-        lines=2, 
-        placeholder="Ej: beber agua, entrenar fuerza, dormir bien...", 
-        label="Tema del consejo"
-    ),
-    outputs=gr.Textbox(lines=8, label="Consejo generado por la AI"),
-    title="Generador de Contenido Fitness AI",
-    description="Introduce un tema de salud o deporte y la IA generará un breve texto o consejo relacionado. Ideal para ideas de redes sociales o newsletters.",
-    article="Modelo basado en GPT-2 Spanish (DeepESP).",
-    submit_btn="Generar Consejo",
-    # El tema Glass es moderno y encaja bien con el estilo 'limpio' de fitness
-    #theme=gr.themes.Glass() 
+    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.",
 )
 
-# 3. Lanzar
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()