Spaces:

GIZ
/

audit_assistant

Running on T4

ppsingh commited on Feb 20

Commit

878b681

verified ·

1 Parent(s): 163d6ef

Update auditqa/process_chunks.py

Files changed (1) hide show

auditqa/process_chunks.py CHANGED Viewed

@@ -61,6 +61,7 @@ def load_chunks():
     # define embedding model
     embeddings = HuggingFaceEmbeddings(
         model_kwargs = {'device': device},
         encode_kwargs = {'normalize_embeddings': bool(int(config.get('retriever','NORMALIZE')))},
         model_name=config.get('retriever','MODEL')
     )

     # define embedding model
     embeddings = HuggingFaceEmbeddings(
         model_kwargs = {'device': device},
+        show_progress= True,
         encode_kwargs = {'normalize_embeddings': bool(int(config.get('retriever','NORMALIZE')))},
         model_name=config.get('retriever','MODEL')
     )