| 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.") | |