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
|