garrethlee commited on
Commit
a99122f
·
verified ·
1 Parent(s): 8bc947f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -39,6 +39,7 @@ def respond(
39
  outputs = pipeline(
40
  messages,
41
  max_new_tokens=max_tokens,
 
42
  temperature=temperature,
43
  top_p=top_p,
44
  eos_token_id=terminators
@@ -56,7 +57,7 @@ with gr.Blocks() as demo:
56
  respond,
57
  additional_inputs=[
58
  gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
59
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
60
  gr.Slider(
61
  minimum=0.1,
62
  maximum=1.0,
 
39
  outputs = pipeline(
40
  messages,
41
  max_new_tokens=max_tokens,
42
+ do_sample = True,
43
  temperature=temperature,
44
  top_p=top_p,
45
  eos_token_id=terminators
 
57
  respond,
58
  additional_inputs=[
59
  gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
60
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
61
  gr.Slider(
62
  minimum=0.1,
63
  maximum=1.0,