Spaces:

jorge-henao
/

ask2democracycol

Runtime error

Jorge Henao commited on Mar 7, 2023

Commit

b9d8bb3

•

1 Parent(s): 6d9f31e

stence transformer update

Files changed (2) hide show

__pycache__/pinecode_quieries.cpython-38.pyc CHANGED Viewed

Binary files a/__pycache__/pinecode_quieries.cpython-38.pyc and b/__pycache__/pinecode_quieries.cpython-38.pyc differ

pinecode_quieries.py CHANGED Viewed

@@ -41,7 +41,7 @@ class PinecodeProposalQueries(DocumentQueries):
         #self.retriever = BM25Retriever(document_store = self.document_store)
         self.retriever = EmbeddingRetriever(
             document_store=self.document_store,
-            embedding_model="multi-qa-MiniLM-L6-dot-v1",
             model_format="sentence_transformers"
         )
         self.document_store.update_embeddings(self.retriever, batch_size=16)

         #self.retriever = BM25Retriever(document_store = self.document_store)
         self.retriever = EmbeddingRetriever(
             document_store=self.document_store,
+            embedding_model="multi-qa-distilbert-dot-v1",
             model_format="sentence_transformers"
         )
         self.document_store.update_embeddings(self.retriever, batch_size=16)