mkoot007/Text2Text

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