Spaces:
Sleeping
Sleeping
File size: 1,267 Bytes
4769745 ec3b425 4769745 ec3b425 4769745 ae2548e 4769745 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import streamlit as st
from transformers import pipeline
# Load the text generation pipeline
generator = pipeline("text-generation")
def generate_bedtime_story(name, interests, friends, age):
story_prompt = f"I want to tell a bed time story to a child named {name}. I want the story to revolve around some of {name}'s main interests, which are {interests}. I also want the story to include some of {name}'s friends, who are called: {friends}. Make sure the story is suitable for a child of the age {age}."
story = generator(story_prompt, max_length=1000)[0]['generated_text']
return story
def main():
st.title("Bedtime Story Generator")
name = st.text_input("Enter the name of the person you're reading to:")
age = st.text_input("Enter the age of the person you're reading to:")
interests = st.text_input("Enter some of things the person is interested in:")
friends = st.text_input("Enter the names of some of their friends:")
if st.button("Generate Story"):
if name and interests and friends:
story = generate_bedtime_story(name, interests, age, friends.split(","))
st.write(story)
else:
st.warning("Please fill in all the inputs.")
if __name__ == "__main__":
main()
|