arsiba commited on
Commit
596e351
·
1 Parent(s): d9ee889

feat: add seperate faiss search field

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -77,8 +77,8 @@ def build_markdown_chunks(docs):
77
  return "\n\n".join(lines)
78
 
79
  @spaces.GPU()
80
- def qa_fn(question, top_k, temperature, max_tokens):
81
- docs, file_sources = retrieve(question, top_k)
82
  file_links = build_markdown_links(file_sources)
83
  markdown_chunks = build_markdown_chunks(docs)
84
  prompt = make_prompt(question, docs)[:8000]
@@ -113,7 +113,8 @@ outputs_chunks = gr.Markdown(label="Used Chunks")
113
  demo = gr.Interface(
114
  fn=qa_fn,
115
  inputs=[
116
- gr.Textbox(lines=2, label="Your Question"),
 
117
  ],
118
  additional_inputs=[
119
  gr.Slider(1, 7, value=4, step=1, label="Top-K Documents"),
 
77
  return "\n\n".join(lines)
78
 
79
  @spaces.GPU()
80
+ def qa_fn(faiss_search, question, top_k, temperature, max_tokens):
81
+ docs, file_sources = retrieve(faiss_search, top_k)
82
  file_links = build_markdown_links(file_sources)
83
  markdown_chunks = build_markdown_chunks(docs)
84
  prompt = make_prompt(question, docs)[:8000]
 
113
  demo = gr.Interface(
114
  fn=qa_fn,
115
  inputs=[
116
+ gr.Textbox(lines=4, label="What Documents are you looking for?"),
117
+ gr.Textbox(lines=1, label="What is your question?"),
118
  ],
119
  additional_inputs=[
120
  gr.Slider(1, 7, value=4, step=1, label="Top-K Documents"),