main fixed draft1

app.py

@@ -1,23 +1,25 @@
-
 from transformers import pipeline
 import gradio as gr
 
-story_generator = pipeline("text2text-generation", model="google/flan-t5-base")
+# Safer model config
+story_generator = pipeline("text2text-generation", model="google/flan-t5-base", tokenizer="google/flan-t5-base")
 
 def generate_story(topic, age_range):
     prompt = f"Write a children’s story for ages {age_range} about {topic}. Use simple language and a friendly tone."
     result = story_generator(prompt, max_length=300, temperature=0.8)
     return result[0]["generated_text"]
 
-iface = gr.Interface(
-    fn=generate_story,
-    inputs=[
-        gr.Textbox(label="Story Topic"),
-        gr.Textbox(label="Age Range (e.g. 4-7)")
-    ],
-    outputs="text",
-    title="Children’s Book Generator",
-    description="Enter a topic and an age range. The AI will generate a simple and friendly children’s story!"
-)
+# Launch only if run directly
+if __name__ == "__main__":
+    iface = gr.Interface(
+        fn=generate_story,
+        inputs=[
+            gr.Textbox(label="Story Topic"),
+            gr.Textbox(label="Age Range (e.g. 4-7)")
+        ],
+        outputs="text",
+        title="Children’s Book Generator",
+        description="Enter a topic and an age range. The AI will generate a simple and friendly children’s story!"
+    )
 
-iface.launch()
+    iface.launch()

requirements.txt

@@ -1,2 +1,4 @@
 transformers
-gradio
+gradio
+sentencepiece
+accelerate