Update app.py

app.py

@@ -1,4 +1,29 @@
+import streamlit as st
 from transformers import pipeline
 
-generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
-generator("EleutherAI has", do_sample=True, min_length=50)
+# 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')
+
+generator = load_model()
+
+# Crear la interfaz de usuario
+st.title("Generador de texto GPT-Neo")
+st.write("Introduce un texto para que el modelo lo continúe:")
+
+# Campo de entrada para el prompt
+user_input = st.text_input("Escribe tu prompt aquí:", "")
+
+# Botón para generar texto
+if st.button("Generar texto"):
+    if user_input:
+        # Generar el texto utilizando el modelo
+        with st.spinner("Generando..."):
+            result = generator(user_input, do_sample=True, min_length=50, max_length=100)
+            generated_text = result[0]['generated_text']
+        # Mostrar el texto generado
+        st.subheader("Texto generado:")
+        st.write(generated_text)
+    else:
+        st.error("Por favor, introduce un prompt antes de generar texto.")