Spaces:
Running
on
T4
Running
on
T4
Update auditqa/doc_process.py
Browse files- auditqa/doc_process.py +2 -2
auditqa/doc_process.py
CHANGED
|
@@ -52,8 +52,8 @@ def process_pdf():
|
|
| 52 |
all_documents[category].append(doc_processed)
|
| 53 |
|
| 54 |
for key, docs_processed in all_documents.items():
|
| 55 |
-
|
| 56 |
-
|
| 57 |
|
| 58 |
embeddings = HuggingFaceEmbeddings(
|
| 59 |
model_kwargs = {'device': device},
|
|
|
|
| 52 |
all_documents[category].append(doc_processed)
|
| 53 |
|
| 54 |
for key, docs_processed in all_documents.items():
|
| 55 |
+
docs_processed = [item for sublist in docs_processed for item in sublist]
|
| 56 |
+
all_documents[key] = docs_processed
|
| 57 |
|
| 58 |
embeddings = HuggingFaceEmbeddings(
|
| 59 |
model_kwargs = {'device': device},
|