Updating the code to use Hugging Face Text Generation pipeline

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
-
-description = "Story generation with GPT-2"
-
-interface = gr.Interface(fn=None, title="Story Generation with GPT-2", inputs=[gr.Textbox(lines=7, label="Story")], description=description, examples=[["Adventurer is approached by a mysterious stranger in the tavern for a new quest"]])
-
+transformers import pipeline
+def generate_story(story):
+  story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
+  generated_story = story_gen("<BOS> <superhero> Batman")
+  return generated_story
+interface = gr.Interface(fn=generate_story, title="Story Generation with GPT-2", inputs=[gr.Textbox(lines=7, label="Story")], description="Story generation with GPT-2", examples=[["Adventurer is approached by a mysterious stranger in the tavern for a new quest"]])
 interface.launch()