Spaces:

GIZ
/

audit_assistant

Running on CPU Upgrade

ppsingh commited on Jul 10

Commit

51456ff

•

1 Parent(s): db1ad09

Update auditqa/engine/vectorstore.py

Files changed (1) hide show

auditqa/engine/vectorstore.py CHANGED Viewed

@@ -6,7 +6,7 @@ from dotenv import load_dotenv
 import os
 provider_retrieval_model = "HF"
-embeddingmodel = "BAAI/bge-small-en-v1.5"
 load_dotenv()
 HF_Token = os.environ.get("HF_TOKEN")
 client_path = f"./vectorstore"
@@ -22,7 +22,7 @@ def create_vectorstore(docs):
           api_key=HF_Token, model_name=embeddingmodel
       )
-      dim = 1024

 import os
 provider_retrieval_model = "HF"
+embeddingmodel = "sentence-transformers/all-MiniLM-l6-v2"
 load_dotenv()
 HF_Token = os.environ.get("HF_TOKEN")
 client_path = f"./vectorstore"
           api_key=HF_Token, model_name=embeddingmodel
       )
+      dim = 384