Ahsen Khaliq commited on
Commit
71b80db
1 Parent(s): 98c17e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ examples = [
10
  ['Paris is the capital of',"gpt2-medium"]
11
  ]
12
 
13
- io1 = gr.Interface.load("huggingface/gpt2")
14
 
15
  io2 = gr.Interface.load("huggingface/gpt2-large")
16
 
@@ -33,7 +33,7 @@ def inference(text, model):
33
 
34
  gr.Interface(
35
  inference,
36
- [gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["gpt2-large","gpt2-medium","gpt2-xl"], type="value", default="gpt2-medium", label="model")
37
  ],
38
  gr.outputs.Textbox(label="Output"),
39
  examples=examples,
10
  ['Paris is the capital of',"gpt2-medium"]
11
  ]
12
 
13
+ io1 = gr.Interface.load("huggingface/distilgpt2")
14
 
15
  io2 = gr.Interface.load("huggingface/gpt2-large")
16
 
33
 
34
  gr.Interface(
35
  inference,
36
+ [gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["distilgpt2","gpt2-medium","gpt2-large","gpt2-xl"], type="value", default="gpt2-medium", label="model")
37
  ],
38
  gr.outputs.Textbox(label="Output"),
39
  examples=examples,