riotu-lab commited on
Commit
829dadb
1 Parent(s): 220125b

update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -33,12 +33,12 @@ model_options = [
33
  inputs_component = [
34
  gr.Dropdown(choices=model_options, label="Select Model"),
35
  gr.Textbox(lines=2, placeholder="Enter your text here...", label="Input Text"),
36
- gr.Slider(minimum=1, maximum=10, step=1, default=5, label="Num Beams"),
37
- gr.Slider(minimum=50, maximum=300, step=10, default=200, label="Max Length"),
38
- gr.Slider(minimum=0.0, maximum=1.0, step=0.1, default=0.9, label="Top p"),
39
- gr.Slider(minimum=0.0, maximum=1.0, step=0.1, default=0.1, label="Temperature"),
40
- gr.Slider(minimum=1.0, maximum=5.0, step=0.5, default=3.0, label="Repetition Penalty"),
41
- gr.Slider(minimum=2, maximum=5, step=1, default=3, label="No Repeat Ngram Size")
42
  ]
43
 
44
  # Setup the interface
 
33
  inputs_component = [
34
  gr.Dropdown(choices=model_options, label="Select Model"),
35
  gr.Textbox(lines=2, placeholder="Enter your text here...", label="Input Text"),
36
+ gr.Slider(minimum=1, maximum=10, step=1, label="Num Beams"),
37
+ gr.Slider(minimum=50, maximum=300, step=10, label="Max Length"),
38
+ gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label="Top p"),
39
+ gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label="Temperature"),
40
+ gr.Slider(minimum=1.0, maximum=5.0, step=0.5, label="Repetition Penalty"),
41
+ gr.Slider(minimum=2, maximum=5, step=1, label="No Repeat Ngram Size")
42
  ]
43
 
44
  # Setup the interface