File size: 586 Bytes
a970736
e91e0ec
5a65659
c805802
 
 
83c780e
5a65659
 
c805802
619e8b5
83c780e
9ee8e29
83c780e
5a65659
 
1cfb225
5a65659
619e8b5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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."]
    ],
	inputs=[gr.Textbox(lines=7, label="Story URL")],
	outputs=[gr.Textbox(lines=7, label="Story Summary")]
)

demo.launch(share=True)