Streamlit writer

app.py

@@ -1,7 +1,15 @@
+import streamlit as st
 from transformers import pipeline 
-generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B") 
 
+# Configuración del generador
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+
+# Definir el prompt
 prompt = "hola"
-result = generator( prompt, max_length=400, num_return_sequences=1, do_sample=True, temperature=0.7, top_p=0.9, repetition_penalty=1.1, ) 
 
-print(result[0]['generated_text'])
+# Generar el texto
+result = generator(prompt, max_length=400, num_return_sequences=1, do_sample=True, temperature=0.7, top_p=0.9, repetition_penalty=1.1)
+
+# Mostrar el resultado en la aplicación Streamlit
+st.write("Texto generado:")
+st.write(result[0]['generated_text'])