Update app.py

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from transformers import pipeline
+
+def generate_story(prompt):
+    pipe = pipeline("text-generation", model="Fathi7ma/tiny-stories-gpt2")
+    story = pipe(
+        prompt,
+        max_length=150,
+        num_return_sequences=1,
+        do_sample=True,
+        temperature=0.8
+    )
+    return story[0]["generated_text"]
+
+demo = gr.Interface(
+    fn=generate_story,
+    inputs=gr.Textbox(label="Enter a story prompt"),
+    outputs=gr.Textbox(label="Generated Story"),
+    title="Tiny Story Generator",
+    description="A fine-tuned GPT-2 model that writes short creative stories."
+)
+
+if __name__ == "__main__":
+    demo.launch()