YC-Chen commited on
Commit
64bf8b7
·
verified ·
1 Parent(s): eed1967

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -34
app.py CHANGED
@@ -95,6 +95,41 @@ def refusal_condition(query):
95
  with gr.Blocks() as demo:
96
  gr.Markdown(DESCRIPTION)
97
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
98
  chatbot = gr.Chatbot()
99
  with gr.Row():
100
  msg = gr.Textbox(
@@ -102,6 +137,7 @@ with gr.Blocks() as demo:
102
  show_label=False,
103
  placeholder='Type a message...',
104
  scale=10,
 
105
  )
106
  submit_button = gr.Button('Submit',
107
  variant='primary',
@@ -115,40 +151,6 @@ with gr.Blocks() as demo:
115
 
116
  saved_input = gr.State()
117
 
118
- with gr.Accordion(label='Advanced options', open=False):
119
- system_prompt = gr.Textbox(label='System prompt',
120
- value=DEFAULT_SYSTEM_PROMPT,
121
- lines=6)
122
- max_new_tokens = gr.Slider(
123
- label='Max new tokens',
124
- minimum=32,
125
- maximum=1024,
126
- step=1,
127
- value=512,
128
- )
129
- temperature = gr.Slider(
130
- label='Temperature',
131
- minimum=0.01,
132
- maximum=1.0,
133
- step=0.01,
134
- value=0.01,
135
- )
136
- top_p = gr.Slider(
137
- label='Top-p (nucleus sampling)',
138
- minimum=0.01,
139
- maximum=0.99,
140
- step=0.01,
141
- value=0.01,
142
- )
143
- repetition_penalty = gr.Slider(
144
- label='Repetition Penalty',
145
- minimum=0.1,
146
- maximum=2,
147
- step=0.01,
148
- value=1.1,
149
- )
150
-
151
-
152
  def user(user_message, history):
153
  return "", history + [[user_message, None]]
154
 
 
95
  with gr.Blocks() as demo:
96
  gr.Markdown(DESCRIPTION)
97
 
98
+ system_prompt = gr.Textbox(label='System prompt',
99
+ value=DEFAULT_SYSTEM_PROMPT,
100
+ lines=6)
101
+
102
+ with gr.Accordion(label='Advanced options', open=False):
103
+
104
+ max_new_tokens = gr.Slider(
105
+ label='Max new tokens',
106
+ minimum=32,
107
+ maximum=1024,
108
+ step=1,
109
+ value=512,
110
+ )
111
+ temperature = gr.Slider(
112
+ label='Temperature',
113
+ minimum=0.01,
114
+ maximum=1.0,
115
+ step=0.01,
116
+ value=0.01,
117
+ )
118
+ top_p = gr.Slider(
119
+ label='Top-p (nucleus sampling)',
120
+ minimum=0.01,
121
+ maximum=0.99,
122
+ step=0.01,
123
+ value=0.01,
124
+ )
125
+ repetition_penalty = gr.Slider(
126
+ label='Repetition Penalty',
127
+ minimum=0.1,
128
+ maximum=2,
129
+ step=0.01,
130
+ value=1.1,
131
+ )
132
+
133
  chatbot = gr.Chatbot()
134
  with gr.Row():
135
  msg = gr.Textbox(
 
137
  show_label=False,
138
  placeholder='Type a message...',
139
  scale=10,
140
+ lines=6
141
  )
142
  submit_button = gr.Button('Submit',
143
  variant='primary',
 
151
 
152
  saved_input = gr.State()
153
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
154
  def user(user_message, history):
155
  return "", history + [[user_message, None]]
156