JimmyK300 commited on
Commit
7cc2e2f
·
verified ·
1 Parent(s): cb2edb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -21,8 +21,8 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
21
  response = model(
22
  prompt,
23
  max_tokens=max_tokens,
24
- temperature=temperature,
25
- top_p=top_p
26
  )
27
 
28
  # Extract text response
@@ -33,9 +33,9 @@ demo = gr.ChatInterface(
33
  respond,
34
  additional_inputs=[
35
  gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
36
- gr.Slider(minimum=10, maximum=512, value=100, step=10, label="Max new tokens"),
37
- gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
38
- gr.Slider(minimum=0.1, maximum=1.0, value=0.75, step=0.05, label="Top-p (nucleus sampling)"),
39
  ],
40
  )
41
 
 
21
  response = model(
22
  prompt,
23
  max_tokens=max_tokens,
24
+ # temperature=temperature,
25
+ # top_p=top_p
26
  )
27
 
28
  # Extract text response
 
33
  respond,
34
  additional_inputs=[
35
  gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
36
+ gr.Slider(minimum=10, maximum=1024, value=256, step=10, label="Max new tokens"),
37
+ gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature no effect"),
38
+ gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p (nucleus sampling) no effect"),
39
  ],
40
  )
41