Jezia commited on
Commit
6ebfe78
·
1 Parent(s): 22a0055

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -28
app.py CHANGED
@@ -1,29 +1,12 @@
1
  import gradio as gr
2
- from gradio import mix
3
-
4
- title = "Miniature GPT"
5
- description = "Gradio Demo for a miniature GPT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
6
-
7
-
8
- examples = [
9
- ['A kid is playing']
10
- ]
11
-
12
- io = gr.Interface.load("keras-io/text-generation-miniature-gpt")
13
-
14
-
15
-
16
- def inference(text):
17
-
18
- return io(text)
19
-
20
-
21
-
22
- gr.Interface(
23
- inference,
24
- [gr.inputs.Textbox(label="Input")],
25
- gr.outputs.Textbox(label="Output"),
26
- examples=examples,
27
- # article=article,
28
- title=title,
29
- description=description).launch()
 
1
  import gradio as gr
2
+ from gradio import inputs
3
+ description = "Story generation with GPT-2"
4
+ interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
5
+ title = "Story Generation with GPT-2",
6
+ inputs = [
7
+ gr.inputs.Textbox(lines=7, label="Story"),
8
+ ],
9
+ description=description,
10
+ examples=[["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
11
+ )
12
+ interface.launch()