Nick088 commited on
Commit
48e61ad
·
verified ·
1 Parent(s): 39f4cf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -24,11 +24,11 @@ def generate(
24
  return better_prompt
25
 
26
  additional_inputs=[
27
- gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens"),
28
- gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty"),
29
- gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature"),
30
- gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P"),
31
- gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K"),
32
  ]
33
 
34
  examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]
 
24
  return better_prompt
25
 
26
  additional_inputs=[
27
+ gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens", info="The maximum numbers of new tokens, controls how long is the output"),
28
+ gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty", info="Penalize repeated tokens, making the AI repeat less itself"),
29
+ gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature", info="Higher values produce more diverse outputs"),
30
+ gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P", info="Higher values sample more low-probability tokens"),
31
+ gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens"),
32
  ]
33
 
34
  examples=[["Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.", None, None ]]