vishwask commited on
Commit
b8bc3c1
·
verified ·
1 Parent(s): a1e50de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -238,11 +238,15 @@ def demo():
238
  vector_db = gr.State()
239
  qa_chain = gr.State()
240
  collection_name = gr.State()
241
- slider_chunk_size = 2048
242
- slider_chunk_overlap = 256
 
 
 
243
  with gr.Row():
244
- document = gr.Files(height=100, file_count="multiple",visible=True,
245
- file_types=["pdf"], interactive=True, label="Upload your PDF documents (single or multiple)")
 
246
  with gr.Row():
247
  db_btn = gr.Radio(["ChromaDB"], label="Vector database type", value = "ChromaDB", type="index", info="Choose your vector database",visible=False)
248
  with gr.Accordion("Advanced options - Document text splitter", open=False, visible=False):
 
238
  vector_db = gr.State()
239
  qa_chain = gr.State()
240
  collection_name = gr.State()
241
+ pdf_directory = '/home/user/app/pdfs'
242
+
243
+ pdf_files = [f for f in os.listdir(pdf_directory) if f.endswith(".pdf")]
244
+ gr_files = [gr.File(os.path.join(pdf_directory, file)) for file in pdf_files]
245
+
246
  with gr.Row():
247
+ # document = gr.Files(value = '/home/user/app/pdfs', height=100, file_count="multiple",visible=True,
248
+ # file_types=["pdf"], interactive=True, label="Upload your PDF documents (single or multiple)")
249
+ document = [gr.File(os.path.join(pdf_directory, file)) for file in pdf_files]
250
  with gr.Row():
251
  db_btn = gr.Radio(["ChromaDB"], label="Vector database type", value = "ChromaDB", type="index", info="Choose your vector database",visible=False)
252
  with gr.Accordion("Advanced options - Document text splitter", open=False, visible=False):