ganeshkamath89's picture
Adding more examples for story generation themes
83c780e verified
raw
history blame
774 Bytes
import gradio as gr
from transformers import pipeline
def generate_story(story):
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
story_expanded = story_gen(story)
return story_expanded['generated_text']
demo = gr.Interface (
fn=generate_story,
description="Story generation with GPT-2",
examples=[
["Adventurer is approached by a mysterious stranger in the tavern for a new quest."],
["He didn't remember the last time he had to run this fast as he jumped on to the bus."],
["Fate has it's own plans for the common man, which is why we philosophise on "]
],
inputs=[gr.Textbox(lines=7, label="Story URL")],
outputs=[gr.Textbox(lines=7, label="Story Summary")]
)
demo.launch(share=True)