dennyaw commited on
Commit
e78e4af
1 Parent(s): 07f2080

update parameter

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -107,9 +107,9 @@ def bot(history,temperature, max_new_tokens, top_p,top_k):
107
 
108
  with gr.Blocks() as demo:
109
  temperature = gr.Slider(0, 5, value=0.8, step=0.1, label='Temperature')
110
- max_length = gr.Slider(0, 8192, value=256, step=1, label='Max Length')
111
  top_p = gr.Slider(0, 1, value=0.8, step=0.1, label='Top P')
112
- top_k = gr.Slider(0, 50, value=50, step=1, label='Top K')
113
 
114
  chatbot = gr.Chatbot()
115
  msg = gr.Textbox()
 
107
 
108
  with gr.Blocks() as demo:
109
  temperature = gr.Slider(0, 5, value=0.8, step=0.1, label='Temperature')
110
+ max_length = gr.Slider(0, 1024, value=50, step=1, label='Max Length')
111
  top_p = gr.Slider(0, 1, value=0.8, step=0.1, label='Top P')
112
+ top_k = gr.Slider(0, 50, value=10, step=1, label='Top K')
113
 
114
  chatbot = gr.Chatbot()
115
  msg = gr.Textbox()