Spaces:
Runtime error
Runtime error
Jorge Henao
commited on
Commit
·
0891da7
1
Parent(s):
7123b4d
pipe fix
Browse files- pinecode_quieries.py +2 -3
pinecode_quieries.py
CHANGED
@@ -44,9 +44,8 @@ class PinecodeProposalQueries(DocumentQueries):
|
|
44 |
embedding_model="multi-qa-MiniLM-L6-cos-v1",
|
45 |
model_format="sentence_transformers"
|
46 |
)
|
47 |
-
self.document_store.update_embeddings(self.retriever, batch_size=16
|
48 |
-
)
|
49 |
-
#self.pipe = ExtractiveQAPipeline(self.reader, self.retriever)
|
50 |
|
51 |
def search_by_query(self, query : str, retriever_top_k: int, reader_top_k: int, es_index: str = None) :
|
52 |
#self.log.write_log(query, "hfspace-informecomision")
|
|
|
44 |
embedding_model="multi-qa-MiniLM-L6-cos-v1",
|
45 |
model_format="sentence_transformers"
|
46 |
)
|
47 |
+
self.document_store.update_embeddings(self.retriever, batch_size=16)
|
48 |
+
self.pipe = ExtractiveQAPipeline(self.reader, self.retriever)
|
|
|
49 |
|
50 |
def search_by_query(self, query : str, retriever_top_k: int, reader_top_k: int, es_index: str = None) :
|
51 |
#self.log.write_log(query, "hfspace-informecomision")
|