IA_IC_SEM02_2 / app.py
sorizano's picture
Update app.py
9b95e7e verified
raw
history blame contribute delete
857 Bytes
import streamlit as st
from transformers import pipeline
# Configuraci贸n de la aplicaci贸n de Streamlit
st.title("Generaci贸n de Texto con Hugging Face")
st.write("Introduce un texto para que el modelo lo complete:")
# Entrada de texto del usuario
user_input = st.text_area("Texto de entrada")
# Cargar el modelo de generaci贸n de texto
text_generator = pipeline("text-generation", model="flax-community/gpt-2-spanish")
# Generar texto basado en la entrada del usuario
if user_input:
generated_text = text_generator(user_input, max_length=50, num_return_sequences=1)
# Mostrar el resultado de la generaci贸n de texto
st.write("Texto generado:")
st.write(generated_text[0]['generated_text'])
# Mensaje de informaci贸n
st.info("Este es un ejemplo b谩sico de generaci贸n de texto utilizando un modelo preentrenado de Hugging Face.")