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