File size: 719 Bytes
67448d5 d2bc37a bc75555 d2bc37a 67448d5 d2bc37a bc75555 d2bc37a 67448d5 d2bc37a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import streamlit as st
from transformers import pipeline, set_seed
# Create a text generation pipeline with the "gpt2" model
pipe = pipeline("text-generation", model="gpt2")
st.title("Poem Generator")
user_word = st.text_input("Enter a word:")
if st.button("Generate Poem"):
if user_word:
# Set a seed for reproducibility
set_seed(42)
# Prompt the model to generate a poem based on the user's word
poem_prompt = f"Write a poem about '{user_word}':"
poem = pipe(poem_prompt, max_length=100, do_sample=True, num_return_sequences=1)[0]["generated_text"]
st.markdown("**Poem:**")
st.markdown(poem)
else:
st.warning("Please enter a word.")
|