Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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},
|