Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def add_pdf_gradio(pdf_file_binary, progress=gr.Progress()):
|
|
52 |
with open(os.path.join(this_cache_dir, f"src.pdf"), 'wb') as file:
|
53 |
file.write(pdf_file_binary)
|
54 |
|
55 |
-
dpi =
|
56 |
doc = fitz.open("pdf", pdf_file_binary)
|
57 |
|
58 |
reps_list = []
|
@@ -211,15 +211,15 @@ Our model is capable of:
|
|
211 |
with gr.Row():
|
212 |
kb_id_input = gr.Text(label="Your Knowledge Base ID (paste your Knowledge Base ID here:)")
|
213 |
query_input = gr.Text(label="Your Queston")
|
214 |
-
topk_input = inputs=gr.Number(value=
|
215 |
-
retrieve_button = gr.Button("Retrieve")
|
216 |
|
217 |
with gr.Row():
|
218 |
downvote_button = gr.Button("🤣Downvote")
|
219 |
upvote_button = gr.Button("🤗Upvote")
|
220 |
|
221 |
with gr.Row():
|
222 |
-
images_output = gr.Gallery(label="Retrieved Pages")
|
223 |
|
224 |
retrieve_button.click(retrieve_gradio, inputs=[kb_id_input, query_input, topk_input], outputs=images_output)
|
225 |
|
|
|
52 |
with open(os.path.join(this_cache_dir, f"src.pdf"), 'wb') as file:
|
53 |
file.write(pdf_file_binary)
|
54 |
|
55 |
+
dpi = 200
|
56 |
doc = fitz.open("pdf", pdf_file_binary)
|
57 |
|
58 |
reps_list = []
|
|
|
211 |
with gr.Row():
|
212 |
kb_id_input = gr.Text(label="Your Knowledge Base ID (paste your Knowledge Base ID here:)")
|
213 |
query_input = gr.Text(label="Your Queston")
|
214 |
+
topk_input = inputs=gr.Number(value=5, minimum=1, maximum=10, step=1, label="Number of pages to retrieve")
|
215 |
+
retrieve_button = gr.Button("Step 1: Retrieve")
|
216 |
|
217 |
with gr.Row():
|
218 |
downvote_button = gr.Button("🤣Downvote")
|
219 |
upvote_button = gr.Button("🤗Upvote")
|
220 |
|
221 |
with gr.Row():
|
222 |
+
images_output = gr.Gallery(label="Step 2: Retrieved Pages")
|
223 |
|
224 |
retrieve_button.click(retrieve_gradio, inputs=[kb_id_input, query_input, topk_input], outputs=images_output)
|
225 |
|