Update app.py

app.py

@@ -4,12 +4,12 @@ from transformers import pipeline
 # Inicializar el modelo generador de texto
 @st.cache_resource  # Esto cachea el modelo para no cargarlo repetidamente
 def load_model():
-    return pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
+    return pipeline('text-generation', model='distilgpt2')  # Cambia el modelo a distilgpt2
 
 generator = load_model()
 
 # Crear la interfaz de usuario
-st.title("Generador de texto GPT-Neo")
+st.title("Generador de texto DistilGPT-2")  # Actualiza el título para reflejar el nuevo modelo
 st.write("Introduce un texto para que el modelo lo continúe:")
 
 # Campo de entrada para el prompt
@@ -34,4 +34,4 @@ if st.button("Generar texto"):
         st.subheader("Texto generado:")
         st.write(generated_text)
     else:
-        st.error("Por favor, introduce un prompt antes de generar texto.")
+        st.error("Por favor, introduce un prompt antes de generar texto.")