Spaces:
Runtime error
Runtime error
Update NBDT_Data_Recs.ipynb
Browse files- NBDT_Data_Recs.ipynb +2 -2
NBDT_Data_Recs.ipynb
CHANGED
@@ -430,7 +430,7 @@
|
|
430 |
"name": "stderr",
|
431 |
"output_type": "stream",
|
432 |
"text": [
|
433 |
-
"WARNING:sentence_transformers.SentenceTransformer:No sentence-transformers model found with name /root/.cache/torch/sentence_transformers/biodatlab_MIReAD-Neuro. Creating a new one with MEAN pooling.\n"
|
434 |
]
|
435 |
},
|
436 |
{
|
@@ -770,7 +770,7 @@
|
|
770 |
"def k_inference(query,k):\n",
|
771 |
" return inference(query,k,'koala')\n",
|
772 |
"\n",
|
773 |
-
"model_name = \"biodatlab/MIReAD-Neuro\"\n",
|
774 |
"model_kwargs = {'device': 'cpu'}\n",
|
775 |
"encode_kwargs = {'normalize_embeddings': False}\n",
|
776 |
"faiss_embedder = HuggingFaceEmbeddings(\n",
|
|
|
430 |
"name": "stderr",
|
431 |
"output_type": "stream",
|
432 |
"text": [
|
433 |
+
"WARNING:sentence_transformers.SentenceTransformer:No sentence-transformers model found with name /root/.cache/torch/sentence_transformers/biodatlab_MIReAD-Neuro-Large. Creating a new one with MEAN pooling.\n"
|
434 |
]
|
435 |
},
|
436 |
{
|
|
|
770 |
"def k_inference(query,k):\n",
|
771 |
" return inference(query,k,'koala')\n",
|
772 |
"\n",
|
773 |
+
"model_name = \"biodatlab/MIReAD-Neuro-Large\"\n",
|
774 |
"model_kwargs = {'device': 'cpu'}\n",
|
775 |
"encode_kwargs = {'normalize_embeddings': False}\n",
|
776 |
"faiss_embedder = HuggingFaceEmbeddings(\n",
|