Jorge Henao commited on
Commit
0891da7
·
1 Parent(s): 7123b4d
Files changed (1) hide show
  1. 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")