JimmyK300 commited on
Commit
d5b892f
·
verified ·
1 Parent(s): b84b7c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -30,9 +30,9 @@ def respond(
30
 
31
  generated_ids = model.generate(
32
  **model_inputs,
33
- max_new_tokens=max_tokens,
34
- temperature=temperature,
35
- top_p=top_p,
36
  )
37
 
38
  response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
@@ -46,9 +46,9 @@ demo = gr.ChatInterface(
46
  respond,
47
  additional_inputs=[
48
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
49
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
50
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
51
- gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
52
  ],
53
  )
54
 
 
30
 
31
  generated_ids = model.generate(
32
  **model_inputs,
33
+ # max_new_tokens=max_tokens,
34
+ # temperature=temperature,
35
+ # top_p=top_p,
36
  )
37
 
38
  response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
46
  respond,
47
  additional_inputs=[
48
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
49
+ # gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
50
+ # gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
51
+ # gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
52
  ],
53
  )
54