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