import streamlit as st | |
from transformers import pipeline | |
# Load the model | |
story_generator = pipeline('text-generation', model='gpt2') | |
st.title('Interactive Story Creator') | |
# Text area for user input | |
user_input = st.text_area("Start your story here...", "Once upon a time") | |
if st.button('Generate Story Continuation'): | |
with st.spinner('AI is writing the story...'): | |
generated_story = story_generator(user_input, max_length=200) | |
st.text_area("Here's the continuation of your story:", generated_story[0]['generated_text'], height=250) | |