YC-Chen commited on
Commit
8971fda
1 Parent(s): b168ade

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -61,24 +61,24 @@ with gr.Blocks() as demo:
61
  lines=6)
62
  max_new_tokens = gr.Slider(
63
  label='Max new tokens',
64
- minimum=1,
65
  maximum=1024,
66
  step=1,
67
  value=512,
68
  )
69
  temperature = gr.Slider(
70
  label='Temperature',
71
- minimum=0.0,
72
  maximum=0.5,
73
- step=0.1,
74
- value=0.0,
75
  )
76
  top_p = gr.Slider(
77
  label='Top-p (nucleus sampling)',
78
- minimum=0.05,
79
  maximum=1.0,
80
- step=0.05,
81
- value=0.95,
82
  )
83
 
84
 
@@ -88,6 +88,9 @@ with gr.Blocks() as demo:
88
 
89
  def bot(history, max_new_tokens, temperature, top_p, system_prompt):
90
  chat_data = []
 
 
 
91
  for user_msg, assistant_msg in history:
92
  if user_msg is not None:
93
  chat_data.append({"role": "user", "content": user_msg})
 
61
  lines=6)
62
  max_new_tokens = gr.Slider(
63
  label='Max new tokens',
64
+ minimum=32,
65
  maximum=1024,
66
  step=1,
67
  value=512,
68
  )
69
  temperature = gr.Slider(
70
  label='Temperature',
71
+ minimum=0.01,
72
  maximum=0.5,
73
+ step=0.01,
74
+ value=0.01,
75
  )
76
  top_p = gr.Slider(
77
  label='Top-p (nucleus sampling)',
78
+ minimum=0.01,
79
  maximum=1.0,
80
+ step=0.01,
81
+ value=0.01,
82
  )
83
 
84
 
 
88
 
89
  def bot(history, max_new_tokens, temperature, top_p, system_prompt):
90
  chat_data = []
91
+ system_prompt = system_prompt.strip()
92
+ if system_prompt:
93
+ chat_data.append({"role": "system", "content": system_prompt})
94
  for user_msg, assistant_msg in history:
95
  if user_msg is not None:
96
  chat_data.append({"role": "user", "content": user_msg})