charoori commited on
Commit
de66964
1 Parent(s): 7f88221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -1,12 +1,13 @@
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()
 
1
  import gradio as gr
2
+
3
+ title = "GPT-J-6B"
4
+ description = "Gradio Demo for GPT-J 6B, a transformer model trained using Ben Wang's Mesh Transformer JAX. 'GPT-J' refers to the class of model, while '6B' represents the number of trainable parameters. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
5
+ article = "<p style='text-align: center'><a href='https://github.com/kingoflolz/mesh-transformer-jax' target='_blank'>GPT-J-6B: A 6 Billion Parameter Autoregressive Language Model</a></p>"
6
+
7
+ gr.Interface.load(
8
+ "huggingface/EleutherAI/gpt-j-6B",
9
+ inputs=gr.Textbox(lines=5, label="Input Text"),
10
+ title=title,
11
  description=description,
12
+ article=article,
13
+ ).launch()