Update app.py
Browse files
app.py
CHANGED
@@ -20,6 +20,7 @@ import torch
|
|
20 |
import tqdm
|
21 |
import accelerate
|
22 |
|
|
|
23 |
|
24 |
|
25 |
# default_persist_directory = './chroma_HF/'
|
@@ -197,7 +198,6 @@ def demo():
|
|
197 |
qa_chain = gr.State()
|
198 |
collection_name = gr.State()
|
199 |
pdf_directory = '/home/user/app/pdfs'
|
200 |
-
llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
|
201 |
|
202 |
def process_pdfs():
|
203 |
# List all PDF files in the directory
|
|
|
20 |
import tqdm
|
21 |
import accelerate
|
22 |
|
23 |
+
llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
|
24 |
|
25 |
|
26 |
# default_persist_directory = './chroma_HF/'
|
|
|
198 |
qa_chain = gr.State()
|
199 |
collection_name = gr.State()
|
200 |
pdf_directory = '/home/user/app/pdfs'
|
|
|
201 |
|
202 |
def process_pdfs():
|
203 |
# List all PDF files in the directory
|