model changed to falcon

app.py

@@ -1,25 +1,26 @@
 from transformers import pipeline
 import gradio as gr
 
-# Safer model config
-story_generator = pipeline("text2text-generation", model="google/flan-t5-base", tokenizer="google/flan-t5-base")
+story_generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
 
+
+# Define the story generation function
 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"]
+    story = story_generator(prompt, max_length=400, do_sample=True, temperature=0.8)
+    return story[0]['generated_text']
+
+# Launch Gradio interface
+iface = gr.Interface(
+    fn=generate_story,
+    inputs=[
+        gr.Textbox(label="Story Topic (e.g., dragons, friendship)"),
+        gr.Textbox(label="Age Range (e.g., 5-7)")
+    ],
+    outputs="text",
+    title="Children’s Book Generator",
+    description="Enter a topic and age range to generate a friendly, age-appropriate 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()