Spaces:
Sleeping
Sleeping
ganeshkamath89
commited on
Commit
•
7402477
1
Parent(s):
eba968d
Adding output as label
Browse files
app.py
CHANGED
@@ -4,5 +4,5 @@ def generate_story(story):
|
|
4 |
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
|
5 |
generated_story = story_gen("<BOS> <superhero> Batman")
|
6 |
return generated_story
|
7 |
-
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"]])
|
8 |
interface.launch()
|
|
|
4 |
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
|
5 |
generated_story = story_gen("<BOS> <superhero> Batman")
|
6 |
return generated_story
|
7 |
+
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"]], outputs="label")
|
8 |
interface.launch()
|