LamaAl commited on
Commit
a649caa
1 Parent(s): 06478fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -39,11 +39,10 @@ description = 'This demo is for a BERT2BERT model trained for single-turn open-d
39
  gr.Interface(fn=generate_response,
40
  inputs=[
41
  gr.inputs.Textbox(),
42
- gr.inputs.Slider(5, 5, step=1, label='Minimum Output Length'),
43
- gr.inputs.Slider(20, 20, step=1, label='Top-K'),
44
- gr.inputs.Slider(0.4, 0.4, step=1, label='Top-P'),
45
- gr.inputs.Slider(0.1,,0.1 step=1, label='Temperature'),
46
-
47
  ],
48
  outputs="text",
49
  title=title,
 
39
  gr.Interface(fn=generate_response,
40
  inputs=[
41
  gr.inputs.Textbox(),
42
+ gr.inputs.Slider(5, 20, step=1, label='Minimum Output Length'),
43
+ gr.inputs.Slider(0, 1000, step=10, label='Top-K'),
44
+ gr.inputs.Slider(0, 1, step=0.1, label='Top-P'),
45
+ gr.inputs.Slider(0, 3, step=0.1, label='Temperature'),
 
46
  ],
47
  outputs="text",
48
  title=title,