Update app.py

app.py

@@ -3,23 +3,23 @@ import torch
 from transformers import pipeline
 
 # Título de la app
-st.title("Generador de Texto con Hugging Face GPT-2")
+st.title("Generador de Texto en Español - Rápido y Optimizado")
 
-# Inicialización del modelo
+# Cargar modelo en español optimizado
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="gpt2")
+    return pipeline("text-generation", model="datificate/gpt2-small-spanish")  # Modelo entrenado en español
 
 generator = load_model()
 
-# Sidebar para controlar parámetros
+# Sidebar para configuración
 st.sidebar.header("Configuración del Modelo")
-max_tokens = st.sidebar.slider("Máximo de tokens", 50, 500, 100, 10)
-temperature = st.sidebar.slider("Temperatura (Creatividad)", 0.1, 1.0, 0.7, 0.1)
+max_tokens = st.sidebar.slider("Máximo de tokens", 20, 150, 60, 10)  # Reducimos max_length
+temperature = st.sidebar.slider("Temperatura (Creatividad)", 0.1, 1.0, 0.5, 0.1)  # Moderada
 
 # Entrada del usuario
 st.subheader("Entrada")
-prompt = st.text_area("Escribe un texto para que GPT-2 continúe:")
+prompt = st.text_area("Escribe un texto para que el modelo continúe (en español):")
 
 # Generar respuesta
 if st.button("Generar Texto"):
@@ -32,5 +32,4 @@ if st.button("Generar Texto"):
         st.subheader("Texto Generado:")
         st.write(response)
 
-# Mensaje adicional
-st.info("Modifica los parámetros en la barra lateral para ajustar la generación de texto.")
+st.info("Este modelo está optimizado para generar texto en español rápidamente.")