RishuD7 commited on
Commit
799c8be
1 Parent(s): 7f80677

added Pool

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -2
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import os
2
  from chromadb.config import Settings
3
 
4
  from langchain.chains import RetrievalQA
@@ -17,7 +16,8 @@ from torch import cuda, bfloat16
17
  import gradio as gr
18
  import gc
19
  import torch
20
-
 
21
 
22
  persist_directory = "db"
23
  source_directory = 'source_documents'
 
 
1
  from chromadb.config import Settings
2
 
3
  from langchain.chains import RetrievalQA
 
16
  import gradio as gr
17
  import gc
18
  import torch
19
+ import glob
20
+ from multiprocessing import Pool
21
 
22
  persist_directory = "db"
23
  source_directory = 'source_documents'
requirements.txt CHANGED
@@ -13,5 +13,4 @@ pypandoc==1.11
13
  langchain==0.0.177
14
  streamlit
15
  sentence_transformers
16
- gradio
17
- PyPDF2==3.0.1
 
13
  langchain==0.0.177
14
  streamlit
15
  sentence_transformers
16
+ gradio