Update app.py

app.py

@@ -48,3 +48,29 @@ def generate_story(title, model_name="gpt2"):
 
     # Return the cleaned and formatted story
     return cleaned_story
+
+# Create the Gradio interface using gr.Interface
+demo = gr.Interface(
+    fn=generate_story,
+    inputs=[
+        gr.Textbox(label="Enter Story Title", placeholder="Type a title here..."),  # Title input
+        gr.Dropdown(choices=[
+            'gpt2',
+            'gpt2-large',
+            'EleutherAI/gpt-neo-2.7B',
+            'EleutherAI/gpt-j-6B',
+            'maldv/badger-writer-llama-3-8b',
+            'gpt-neo-2.7B'
+        ], value='gpt2', label="Choose Model")  # Model selection
+    ],
+    outputs="text",
+    title="AI Story Generator",
+    description="Generate a creative story using different AI models.",
+    examples=[
+        ["Sara burst into her friend's house, only to find it plunged into darkness. A strange, pulsing glow flickered from the corner, casting eerie shadows on the walls. Her heart raced as she called out, but there was no answer. Something wasn’t right. On the table sat an unfamiliar, glowing device—humming with energy. With a deep breath, Sara stepped closer, knowing that once she touched it, there would be no turning back."]
+    ]
+)
+
+# Launch the interface with sharing enabled
+demo.launch(share=True)
+