CCockrum commited on
Commit
363ecd9
·
verified ·
1 Parent(s): 2bfec82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -54,15 +54,16 @@ def respond(
54
 
55
  with gr.Blocks(css=css) as demo:
56
  with gr.Column(elem_id="chat-panel"):
 
 
 
 
 
 
57
  gr.ChatInterface(
58
  respond,
59
- additional_inputs=[
60
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
61
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
62
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
63
- gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
64
- ]
65
  )
66
 
67
  if __name__ == "__main__":
68
- demo.launch()
 
54
 
55
  with gr.Blocks(css=css) as demo:
56
  with gr.Column(elem_id="chat-panel"):
57
+ with gr.Accordion("⚙️ Settings", open=False):
58
+ system_message = gr.Textbox(value="You are a friendly Chatbot.", label="System message")
59
+ max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
60
+ temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
61
+ top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
62
+
63
  gr.ChatInterface(
64
  respond,
65
+ additional_inputs=[system_message, max_tokens, temperature, top_p]
 
 
 
 
 
66
  )
67
 
68
  if __name__ == "__main__":
69
+ demo.launch()