vaariis commited on
Commit
d9ed99c
1 Parent(s): 0559eba

update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -1,5 +1,11 @@
1
  import gradio as gr
2
- description = "Story generation with GPT"
3
- examples = [["An adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
4
- demo = gr.Interface.load("models/EleutherAI/gpt-neo-1.3B", description=description, examples=examples)
5
- demo.launch()
 
 
 
 
 
 
 
1
  import gradio as gr
2
+
3
+ title = "GPT-J-6B"
4
+ description = "GPT-J 6B, a transformer model trained using Ben Wang's Mesh Transformer JAX.'6B' is the number of trainable parameters. Add your text, or click one of the examples to load them."
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
+ examples = [
7
+ ['A space ranger encounters a strange silhouette.'],
8
+ ["A day on Saturn is 10 hours and 14 minutes."],
9
+ ["There's no oxygen on Saturn, but roughly 75% hydrogen and 25% helium."]
10
+ ]
11
+ gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", inputs=gr.inputs.Textbox(lines=5, label="Input Text"),title=title,description=description,article=article, examples=examples,enable_queue=True).launch()