Spaces:
Running
on
Zero
Running
on
Zero
feat: add seperate faiss search field
Browse files
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(
|
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=
|
|
|
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"),
|