BICORP commited on
Commit
cd515d7
·
verified ·
1 Parent(s): 6fc8750

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,6 +18,7 @@ def chat_with_model(message, history, system_message, max_tokens, temperature, t
18
 
19
  messages = [{"role": "system", "content": system_message}]
20
 
 
21
  for user_msg, assistant_msg in history:
22
  if user_msg:
23
  messages.append({"role": "user", "content": user_msg})
@@ -50,7 +51,7 @@ demo = gr.ChatInterface(
50
  chat_with_model,
51
  additional_inputs=[
52
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
53
- gr.Slider(minimum=1, maximum=1024, value=512, step=1, label="Max new tokens"),
54
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
55
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
56
  gr.Dropdown(lake_models, label="Select Model", value=lake_models[0])
 
18
 
19
  messages = [{"role": "system", "content": system_message}]
20
 
21
+ # Append history messages
22
  for user_msg, assistant_msg in history:
23
  if user_msg:
24
  messages.append({"role": "user", "content": user_msg})
 
51
  chat_with_model,
52
  additional_inputs=[
53
  gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
54
+ gr.Slider(minimum=1, maximum=1024, value=1024, step=1, label="Max new tokens"),
55
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
56
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
57
  gr.Dropdown(lake_models, label="Select Model", value=lake_models[0])