ganeshkamath89 commited on
Commit
7402477
1 Parent(s): eba968d

Adding output as label

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()