Cudd1es commited on
Commit
489cfae
·
1 Parent(s): d3b3aa2

updated app

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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[:3], collection)
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知识问答助手, 只能基于提供资料内容作答。", 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,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()