Arthur Samuel Galego Panucci FIgueiredo commited on
Commit
233bfa8
·
verified ·
1 Parent(s): 4ab42a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -56,7 +56,7 @@ def chat(user_input, system_prompt, temperature, top_p, max_tokens):
56
  do_sample=True,
57
  temperature=float(temperature),
58
  top_p=float(top_p),
59
- repetition_penalty=1.1,
60
  eos_token_id=tokenizer.eos_token_id,
61
  pad_token_id=tokenizer.pad_token_id
62
  )
@@ -119,15 +119,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
119
  )
120
 
121
  temperature = gr.Slider(
122
- 0.1, 1.5, value=0.7, step=0.05, label="Temperature"
123
  )
124
 
125
  top_p = gr.Slider(
126
- 0.1, 1.0, value=0.9, step=0.05, label="Top-p"
127
  )
128
 
129
  max_tokens = gr.Slider(
130
- 50, 512, value=200, step=10, label="Max tokens"
131
  )
132
 
133
  submit.click(
 
56
  do_sample=True,
57
  temperature=float(temperature),
58
  top_p=float(top_p),
59
+ repetition_penalty=1.2,
60
  eos_token_id=tokenizer.eos_token_id,
61
  pad_token_id=tokenizer.pad_token_id
62
  )
 
119
  )
120
 
121
  temperature = gr.Slider(
122
+ 0.1, 1.5, value=0.65, step=0.05, label="Temperature"
123
  )
124
 
125
  top_p = gr.Slider(
126
+ 0.1, 1.0, value=0.95, step=0.05, label="Top-p"
127
  )
128
 
129
  max_tokens = gr.Slider(
130
+ 50, 1024, value=200, step=10, label="Max tokens"
131
  )
132
 
133
  submit.click(