schuler commited on
Commit
4329549
·
verified ·
1 Parent(s): 43c5e78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -57,9 +57,9 @@ def respond(
57
  response_output = generator(
58
  prompt,
59
  generation_config=generator_conf,
60
- max_new_tokens=64,
61
  do_sample=True,
62
- top_p=0.25,
63
  repetition_penalty=1.2
64
  )
65
 
@@ -99,12 +99,12 @@ demo = gr.ChatInterface(
99
  respond,
100
  additional_inputs=[
101
  gr.Textbox(value="You are a friendly Chatbot." + global_error, label="System message"),
102
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
103
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
104
  gr.Slider(
105
  minimum=0.1,
106
  maximum=1.0,
107
- value=0.95,
108
  step=0.05,
109
  label="Top-p (nucleus sampling)",
110
  ),
 
57
  response_output = generator(
58
  prompt,
59
  generation_config=generator_conf,
60
+ max_new_tokens=max_tokens,
61
  do_sample=True,
62
+ top_p=top_p,
63
  repetition_penalty=1.2
64
  )
65
 
 
99
  respond,
100
  additional_inputs=[
101
  gr.Textbox(value="You are a friendly Chatbot." + global_error, label="System message"),
102
+ gr.Slider(minimum=1, maximum=2048, value=64, step=1, label="Max new tokens"),
103
+ # gr.Slider(minimum=0.1, maximum=4.0, value=1.0, step=0.1, label="Temperature"),
104
  gr.Slider(
105
  minimum=0.1,
106
  maximum=1.0,
107
+ value=0.25,
108
  step=0.05,
109
  label="Top-p (nucleus sampling)",
110
  ),