initial commit1

app.py

@@ -1,12 +1,17 @@
+from transformers import pipeline
 import gradio as gr
 
-description = "Story generation with GPT-2"
-title = "Generate your own story"
-examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
+generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
 
-interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
-            description=description,
-            examples=examples
-)
+def generate_story(prompt):
+    result = generator(prompt, max_length=200, do_sample=True)
+    return result[0]["generated_text"]
 
-interface.launch()
+gr.Interface(
+    fn=generate_story,
+    inputs=gr.Textbox(label="Prompt"),
+    outputs=gr.Textbox(label="Generated Story"),
+    title="Generate your own story",
+    description="Story generation with GPT-2",
+    examples=[["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]],
+).launch()