Spaces:
Runtime error
Runtime error
| 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) |