update
Browse files
README.md
CHANGED
|
@@ -58,10 +58,10 @@ A simple root-level script is available for running retrieval directly from a pi
|
|
| 58 |
python main.py --query "what is diabetes" --top-k 3
|
| 59 |
```
|
| 60 |
|
| 61 |
-
By default, the script loads `
|
| 62 |
|
| 63 |
```bash
|
| 64 |
-
python main.py --path path/to/
|
| 65 |
```
|
| 66 |
|
| 67 |
## Deployment Notes for Hugging Face Spaces
|
|
|
|
| 58 |
python main.py --query "what is diabetes" --top-k 3
|
| 59 |
```
|
| 60 |
|
| 61 |
+
By default, the script loads `rag_store_cpu.pkl` from the repository root. To specify a different store file:
|
| 62 |
|
| 63 |
```bash
|
| 64 |
+
python main.py --path path/to/rag_store_cpu.pkl --query "what is diabetes"
|
| 65 |
```
|
| 66 |
|
| 67 |
## Deployment Notes for Hugging Face Spaces
|
main.py
CHANGED
|
@@ -4,9 +4,9 @@ from pathlib import Path
|
|
| 4 |
import torch
|
| 5 |
from sentence_transformers import SentenceTransformer, util
|
| 6 |
|
| 7 |
-
DEFAULT_PICKLE = Path(__file__).parent / "
|
| 8 |
DEFAULT_MODEL = "BAAI/bge-large-en-v1.5"
|
| 9 |
-
GOOGLE_DRIVE_FILE_ID = "
|
| 10 |
GOOGLE_DRIVE_DOWNLOAD_URL = "https://drive.google.com/uc?export=download"
|
| 11 |
|
| 12 |
|
|
|
|
| 4 |
import torch
|
| 5 |
from sentence_transformers import SentenceTransformer, util
|
| 6 |
|
| 7 |
+
DEFAULT_PICKLE = Path(__file__).parent / "rag_store_cpu.pkl"
|
| 8 |
DEFAULT_MODEL = "BAAI/bge-large-en-v1.5"
|
| 9 |
+
GOOGLE_DRIVE_FILE_ID = "1RLtLARA0G0v51CQckpQqfXeD10RSSA39"
|
| 10 |
GOOGLE_DRIVE_DOWNLOAD_URL = "https://drive.google.com/uc?export=download"
|
| 11 |
|
| 12 |
|