ppsingh commited on
Commit
347dbcf
1 Parent(s): 5ded842

Update auditqa/doc_process.py

Browse files
Files changed (1) hide show
  1. 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
- 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},
 
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},