Spaces:
Runtime error
Runtime error
Jorge Henao
commited on
Commit
•
04ca423
1
Parent(s):
012aca1
fixing params
Browse files
__pycache__/config.cpython-38.pyc
CHANGED
|
Binary files a/__pycache__/config.cpython-38.pyc and b/__pycache__/config.cpython-38.pyc differ
|
|
|
__pycache__/pinecode_quieries.cpython-38.pyc
CHANGED
|
Binary files a/__pycache__/pinecode_quieries.cpython-38.pyc and b/__pycache__/pinecode_quieries.cpython-38.pyc differ
|
|
|
app_pinecode.py
CHANGED
|
@@ -26,7 +26,7 @@ def search(question, retriever_top_k, reader_top_k):
|
|
| 26 |
for i in range(0, len(query_result)):
|
| 27 |
item = query_result[i]
|
| 28 |
#result.append([[i+1], item.answer, item.context[:200], item.meta['title'], item.meta['source'], int(item.meta['page'])])
|
| 29 |
-
result.append([[i+1], item.answer, item.context[:200], item.meta['
|
| 30 |
|
| 31 |
return result
|
| 32 |
|
|
|
|
| 26 |
for i in range(0, len(query_result)):
|
| 27 |
item = query_result[i]
|
| 28 |
#result.append([[i+1], item.answer, item.context[:200], item.meta['title'], item.meta['source'], int(item.meta['page'])])
|
| 29 |
+
result.append([[i+1], item.answer, item.context[:200], item.meta['title']])
|
| 30 |
|
| 31 |
return result
|
| 32 |
|
pinecode_quieries.py
CHANGED
|
@@ -41,7 +41,7 @@ class PinecodeProposalQueries(DocumentQueries):
|
|
| 41 |
#self.retriever = BM25Retriever(document_store = self.document_store)
|
| 42 |
self.retriever = EmbeddingRetriever(
|
| 43 |
document_store=self.document_store,
|
| 44 |
-
embedding_model="multi-qa-
|
| 45 |
model_format="sentence_transformers"
|
| 46 |
)
|
| 47 |
self.document_store.update_embeddings(self.retriever, batch_size=16)
|
|
|
|
| 41 |
#self.retriever = BM25Retriever(document_store = self.document_store)
|
| 42 |
self.retriever = EmbeddingRetriever(
|
| 43 |
document_store=self.document_store,
|
| 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)
|