Spaces:

peter2000
/

policy_test

Runtime error

peter2000 commited on Nov 19, 2022

Commit

027bf06

1 Parent(s): 4b71369

Update appStore/keyword_search.py

Files changed (1) hide show

appStore/keyword_search.py CHANGED Viewed

@@ -34,7 +34,7 @@ def start_haystack(documents_processed):
     retriever = EmbeddingRetriever(
                 document_store=document_store,
-                embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
                 model_format="sentence_transformers")
     document_store.update_embeddings(retriever)

     retriever = EmbeddingRetriever(
                 document_store=document_store,
+                embedding_model="peter2000/sdg_sentence_transformer", #"sentence-transformers/multi-qa-mpnet-base-dot-v1",
                 model_format="sentence_transformers")
     document_store.update_embeddings(retriever)