from transformers import pipeline
import gradio as gr

generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")

def generate_story(prompt):
    result = generator(prompt, max_length=200, do_sample=True)
    return result[0]["generated_text"]

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()