Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline, set_seed | |
text_generator = pipeline("text-generation", model="google/flan-t5-base") | |
st.title("Story Generator") | |
title = st.text_input("Title of the Story:") | |
character = st.text_input("Character (Hero) Name:") | |
era = st.selectbox("Era of the Story:", ["Medieval", "Modern", "Future"]) | |
genre = st.selectbox("Genre:", ["Action", "Novelistic", "Love"]) | |
ending = st.selectbox("Ending:", ["Happy", "Sad"]) | |
if st.button("Generate Story"): | |
if title and character: | |
prompt = f"In the {era} era, there was a character named {character}. This is the {genre} story of {character}, a {genre} story with a {ending} ending." | |
set_seed(42) | |
story = text_generator(prompt, max_length=300, do_sample=True)[0]["generated_text"] | |
full_story = f"# {title}\n\n{story}" | |
st.markdown(full_story) | |
else: | |
st.warning("Please provide a title and character name.") | |