ppsingh commited on
Commit
24a7885
1 Parent(s): 2d17ff2

Update auditqa/doc_process.py

Browse files
Files changed (1) hide show
  1. auditqa/doc_process.py +3 -4
auditqa/doc_process.py CHANGED
@@ -43,16 +43,15 @@ def process_pdf():
43
  all_documents[file] = doc_processed
44
 
45
  print(all_documents.keys())
46
- print(all_documents['ABC'])
47
-
48
-
49
 
50
- embeddings = HuggingFaceEmbeddings(
51
  model_name="sentence-transformers/all-mpnet-base-v2"
52
  )
53
 
54
  qdrant_collections = {}
55
  for file,value in all_documents.items():
 
56
  qdrant_collections[file] = Qdrant.from_documents(
57
  value,
58
  embeddings,
 
43
  all_documents[file] = doc_processed
44
 
45
  print(all_documents.keys())
46
+
 
 
47
 
48
+ embeddings = HuggingFaceInferenceAPIEmbeddings(api_key=HF_token,
49
  model_name="sentence-transformers/all-mpnet-base-v2"
50
  )
51
 
52
  qdrant_collections = {}
53
  for file,value in all_documents.items():
54
+ print("emebddings for:",file)
55
  qdrant_collections[file] = Qdrant.from_documents(
56
  value,
57
  embeddings,