on1onmangoes commited on
Commit
5bcfb6f
·
verified ·
1 Parent(s): 0896546

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +69 -71
app.py CHANGED
@@ -162,77 +162,75 @@ with gr.Blocks(css=CSS) as demo:
162
  gr.HTML(TITLE)
163
  with gr.Tab("Chat"):
164
  chatbot = gr.Chatbot() # Create a chatbot interface
165
- gr.Dropdown(client_name,value="primo",label="Select Client Document",allow_custom_value=True),
166
-
167
- # chat_interface = gr.ChatInterface(
168
- # fn=stream_chat_with_rag,
169
- # chatbot=chatbot,
170
- # additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
171
- # additional_inputs=[
172
- # gr.Dropdown(client_name,value="primo",label="Select Client", render=False,allow_custom_value=True),
173
- # gr.Textbox(
174
- # value="You are an expert assistant",
175
- # label="System Prompt",
176
- # render=False,
177
- # ),
178
- # gr.Slider(
179
- # minimum=1,
180
- # maximum=10,
181
- # step=1,
182
- # value=10,
183
- # label="Number of Initial Documents to Retrieve",
184
- # render=False,
185
- # ),
186
- # gr.Slider(
187
- # minimum=1,
188
- # maximum=10,
189
- # step=1,
190
- # value=9,
191
- # label="Number of Final Documents to Retrieve",
192
- # render=False,
193
- # ),
194
- # gr.Slider(
195
- # minimum=0.2,
196
- # maximum=1,
197
- # step=0.1,
198
- # value=0,
199
- # label="Temperature",
200
- # render=False,
201
- # ),
202
- # gr.Slider(
203
- # minimum=128,
204
- # maximum=8192,
205
- # step=1,
206
- # value=1024,
207
- # label="Max new tokens",
208
- # render=False,
209
- # ),
210
- # gr.Slider(
211
- # minimum=0.0,
212
- # maximum=1.0,
213
- # step=0.1,
214
- # value=1.0,
215
- # label="Top P",
216
- # render=False,
217
- # ),
218
- # gr.Slider(
219
- # minimum=1,
220
- # maximum=20,
221
- # step=1,
222
- # value=20,
223
- # label="Top K",
224
- # render=False,
225
- # ),
226
- # gr.Slider(
227
- # minimum=0.0,
228
- # maximum=2.0,
229
- # step=0.1,
230
- # value=1.2,
231
- # label="Repetition Penalty",
232
- # render=False,
233
- # ),
234
- # ],
235
- # )
236
 
237
 
238
  with gr.Tab("Process PDF"):
 
162
  gr.HTML(TITLE)
163
  with gr.Tab("Chat"):
164
  chatbot = gr.Chatbot() # Create a chatbot interface
165
+ chat_interface = gr.ChatInterface(
166
+ fn=stream_chat_with_rag,
167
+ chatbot=chatbot,
168
+ additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
169
+ additional_inputs=[
170
+ gr.Dropdown(client_name,value="primo",label="Select Client", render=False,allow_custom_value=True),
171
+ gr.Textbox(
172
+ value="You are an expert assistant",
173
+ label="System Prompt",
174
+ render=False,
175
+ ),
176
+ gr.Slider(
177
+ minimum=1,
178
+ maximum=10,
179
+ step=1,
180
+ value=10,
181
+ label="Number of Initial Documents to Retrieve",
182
+ render=False,
183
+ ),
184
+ gr.Slider(
185
+ minimum=1,
186
+ maximum=10,
187
+ step=1,
188
+ value=9,
189
+ label="Number of Final Documents to Retrieve",
190
+ render=False,
191
+ ),
192
+ gr.Slider(
193
+ minimum=0.2,
194
+ maximum=1,
195
+ step=0.1,
196
+ value=0,
197
+ label="Temperature",
198
+ render=False,
199
+ ),
200
+ gr.Slider(
201
+ minimum=128,
202
+ maximum=8192,
203
+ step=1,
204
+ value=1024,
205
+ label="Max new tokens",
206
+ render=False,
207
+ ),
208
+ gr.Slider(
209
+ minimum=0.0,
210
+ maximum=1.0,
211
+ step=0.1,
212
+ value=1.0,
213
+ label="Top P",
214
+ render=False,
215
+ ),
216
+ gr.Slider(
217
+ minimum=1,
218
+ maximum=20,
219
+ step=1,
220
+ value=20,
221
+ label="Top K",
222
+ render=False,
223
+ ),
224
+ gr.Slider(
225
+ minimum=0.0,
226
+ maximum=2.0,
227
+ step=0.1,
228
+ value=1.2,
229
+ label="Repetition Penalty",
230
+ render=False,
231
+ ),
232
+ ],
233
+ )
 
 
234
 
235
 
236
  with gr.Tab("Process PDF"):