File size: 602 Bytes
67448d5 8e65c11 03be16d 67448d5 03be16d bc75555 03be16d bb91b0e efa6769 bb91b0e 03be16d efa6769 bb91b0e 67448d5 03be16d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import streamlit as st
from transformers import pipeline
pipe = pipeline("text-generation", model="ismaelfaro/gpt2-poems.en")
st.title("Text Generation with Hugging Face Transformers")
seed_text = st.text_input("Enter a seed text:")
if st.button("Generate Text"):
if seed_text:
generated_text = pipe(seed_text, max_length=100, do_sample=True)[0]['generated_text']
st.write("Generated Text:")
cleaned_text = generated_text.replace('\n', ' ')
st.write(cleaned_text)
else:
st.write("Please enter a seed text to generate text.")
|