willcat commited on
Commit
9d16267
·
1 Parent(s): ce58eb6

add requirements

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -1,12 +1,5 @@
1
- import gradio as gr
 
2
 
3
- description = "Story generation with GPT-2"
4
- title = "Generate your own story"
5
- examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
6
-
7
- interface = gr.load("huggingface/pranavpsv/gpt2-genre-story-generator",
8
- description=description,
9
- examples=examples
10
- )
11
-
12
- interface.launch()
 
1
+ # Example of usage
2
+ from transformers import pipeline
3
 
4
+ story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
5
+ print(story_gen("<BOS> <superhero> Batman"))