Spaces:
Running
Running
rename parameter
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def add_text(history, text):
|
|
34 |
return history, gr.Textbox(value="", interactive=False)
|
35 |
|
36 |
|
37 |
-
def bot(history, api_kind, chunk_table, embedding_model, llm_model,
|
38 |
top_k_param = int(top_k_param)
|
39 |
query = history[-1][0]
|
40 |
|
@@ -119,7 +119,7 @@ with gr.Blocks() as demo:
|
|
119 |
value="gpt-3.5-turbo",
|
120 |
label='LLM'
|
121 |
)
|
122 |
-
|
123 |
choices=[
|
124 |
"None"
|
125 |
"BAAI/bge-reranker-large",
|
@@ -143,14 +143,14 @@ with gr.Blocks() as demo:
|
|
143 |
prompt_html = gr.HTML()
|
144 |
# Turn off interactivity while generating if you click
|
145 |
txt_msg = txt_btn.click(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
|
146 |
-
bot, [chatbot, api_kind, chunk_table, embedding_model, llm_model,
|
147 |
|
148 |
# Turn it back on
|
149 |
txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
|
150 |
|
151 |
# Turn off interactivity while generating if you hit enter
|
152 |
txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
|
153 |
-
bot, [chatbot, api_kind, chunk_table, embedding_model, llm_model,
|
154 |
|
155 |
# Turn it back on
|
156 |
txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
|
|
|
34 |
return history, gr.Textbox(value="", interactive=False)
|
35 |
|
36 |
|
37 |
+
def bot(history, api_kind, chunk_table, embedding_model, llm_model, cross_encoder, top_k_param):
|
38 |
top_k_param = int(top_k_param)
|
39 |
query = history[-1][0]
|
40 |
|
|
|
119 |
value="gpt-3.5-turbo",
|
120 |
label='LLM'
|
121 |
)
|
122 |
+
cross_encoder = gr.Radio(
|
123 |
choices=[
|
124 |
"None"
|
125 |
"BAAI/bge-reranker-large",
|
|
|
143 |
prompt_html = gr.HTML()
|
144 |
# Turn off interactivity while generating if you click
|
145 |
txt_msg = txt_btn.click(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
|
146 |
+
bot, [chatbot, api_kind, chunk_table, embedding_model, llm_model, cross_encoder, top_k_param], [chatbot, prompt_html])
|
147 |
|
148 |
# Turn it back on
|
149 |
txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
|
150 |
|
151 |
# Turn off interactivity while generating if you hit enter
|
152 |
txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
|
153 |
+
bot, [chatbot, api_kind, chunk_table, embedding_model, llm_model, cross_encoder, top_k_param], [chatbot, prompt_html])
|
154 |
|
155 |
# Turn it back on
|
156 |
txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
|