Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 
 # --- CONFIGURACIÓN ---
-MODEL_ID = "datificate/gpt2-medium-spanish"  # modelo más grande
+MODEL_ID = "datificate/gpt2-small-spanish"  # modelo pequeño, seguro en CPU Free
 
 # Crear pipeline de generación de texto
 try:
@@ -17,7 +17,7 @@ except:
 # Lista para guardar historial de la conversación
 historial = []
 
-# Función para generar texto con memoria
+# Función para generar texto con memoria y parámetros optimizados
 def generar_texto(entrada):
     global historial
     if texto_modelo:
@@ -29,10 +29,10 @@ def generar_texto(entrada):
             prompt,
             max_new_tokens=100,
             do_sample=True,
-            temperature=0.8,
-            top_k=40,
-            top_p=0.9,
-            repetition_penalty=1.2
+            temperature=0.8,         # creatividad
+            top_k=40,                # opciones limitadas
+            top_p=0.9,               # probabilidad acumulativa
+            repetition_penalty=1.3   # evita repeticiones
         )
         # Obtenemos solo la respuesta generada
         texto_generado = resultados[0]["generated_text"]
@@ -45,7 +45,7 @@ def generar_texto(entrada):
 
 # --- INTERFAZ GRADIO ---
 with gr.Blocks() as demo:
-    gr.Markdown("## Almanac (GPT-2 Medium, conversación con memoria) 😎")
+    gr.Markdown("## Almanac (GPT-2 Small, conversación optimizada) 😎")
 
     entrada_texto = gr.Textbox(label="Escribe algo")
     salida_texto = gr.Textbox(label="Respuesta de la IA")