updated app
Browse files
app.py
CHANGED
|
@@ -85,7 +85,7 @@ def respond(
|
|
| 85 |
|
| 86 |
total_reranked = reranked + reformulated_reranked
|
| 87 |
deduped = dedup_by_chapter_event(total_reranked, max_per_group=1)
|
| 88 |
-
expanded_results = expand_with_neighbors(deduped[:
|
| 89 |
context = []
|
| 90 |
for idx, text in enumerate(expanded_results):
|
| 91 |
context.append(text[0]) if text else ""
|
|
@@ -124,7 +124,7 @@ chatbot = gr.ChatInterface(
|
|
| 124 |
respond,
|
| 125 |
type="messages",
|
| 126 |
additional_inputs=[
|
| 127 |
-
gr.Textbox(value="你是BangDream知识问答助手,
|
| 128 |
gr.Slider(minimum=64, maximum=8192, value=1536, step=1, label="Max new tokens"),
|
| 129 |
gr.Slider(minimum=0.1, maximum=2.0, value=0.2, step=0.05, label="Temperature"),
|
| 130 |
gr.Slider(
|
|
@@ -147,6 +147,7 @@ with gr.Blocks(title="Dr-Bang RAG QA") as demo:
|
|
| 147 |
with gr.Sidebar():
|
| 148 |
gr.Markdown(
|
| 149 |
"## Dr-Bang QA\n\n"
|
|
|
|
| 150 |
)
|
| 151 |
gr.LoginButton()
|
| 152 |
chatbot.render()
|
|
|
|
| 85 |
|
| 86 |
total_reranked = reranked + reformulated_reranked
|
| 87 |
deduped = dedup_by_chapter_event(total_reranked, max_per_group=1)
|
| 88 |
+
expanded_results = expand_with_neighbors(deduped[:5], collection)
|
| 89 |
context = []
|
| 90 |
for idx, text in enumerate(expanded_results):
|
| 91 |
context.append(text[0]) if text else ""
|
|
|
|
| 124 |
respond,
|
| 125 |
type="messages",
|
| 126 |
additional_inputs=[
|
| 127 |
+
gr.Textbox(value="你是BangDream知识问答助手, 只能基于提供资料内容作答。\n\n[GitHub repo](https://github.com/Cudd1es/dr-bang?tab=readme-ov-file)", label="System message"),
|
| 128 |
gr.Slider(minimum=64, maximum=8192, value=1536, step=1, label="Max new tokens"),
|
| 129 |
gr.Slider(minimum=0.1, maximum=2.0, value=0.2, step=0.05, label="Temperature"),
|
| 130 |
gr.Slider(
|
|
|
|
| 147 |
with gr.Sidebar():
|
| 148 |
gr.Markdown(
|
| 149 |
"## Dr-Bang QA\n\n"
|
| 150 |
+
"[GitHub project](https://github.com/Cudd1es/dr-bang?tab=readme-ov-file)\n\n"
|
| 151 |
)
|
| 152 |
gr.LoginButton()
|
| 153 |
chatbot.render()
|