gholap310 commited on
Commit
cc07c89
·
verified ·
1 Parent(s): 701c2ae

initial commit

Browse files
Files changed (3) hide show
  1. README.md +5 -13
  2. app.py +23 -0
  3. requirements.txt +2 -0
README.md CHANGED
@@ -1,14 +1,6 @@
1
- ---
2
- title: Children Book Generator
3
- emoji: 😻
4
- colorFrom: red
5
- colorTo: green
6
- sdk: gradio
7
- sdk_version: 5.38.2
8
- app_file: app.py
9
- pinned: false
10
- license: mit
11
- short_description: Generate Children book
12
- ---
13
 
14
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
 
2
+ # Children's Book Generator
3
+
4
+ A simple AI app that generates short stories for children based on your topic and age range input.
5
+
6
+ Built using Hugging Face Transformers and Gradio.
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ from transformers import pipeline
3
+ import gradio as gr
4
+
5
+ story_generator = pipeline("text2text-generation", model="google/flan-t5-base")
6
+
7
+ def generate_story(topic, age_range):
8
+ prompt = f"Write a children’s story for ages {age_range} about {topic}. Use simple language and a friendly tone."
9
+ result = story_generator(prompt, max_length=300, temperature=0.8)
10
+ return result[0]["generated_text"]
11
+
12
+ iface = gr.Interface(
13
+ fn=generate_story,
14
+ inputs=[
15
+ gr.Textbox(label="Story Topic"),
16
+ gr.Textbox(label="Age Range (e.g. 4-7)")
17
+ ],
18
+ outputs="text",
19
+ title="Children’s Book Generator",
20
+ description="Enter a topic and an age range. The AI will generate a simple and friendly children’s story!"
21
+ )
22
+
23
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ transformers
2
+ gradio