chintu4 commited on
Commit
4d7f4b5
·
1 Parent(s): f38e6e6
Files changed (2) hide show
  1. README.md +2 -2
  2. main.py +2 -2
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 `rag_store.pkl` from the repository root. To specify a different store file:
62
 
63
  ```bash
64
- python main.py --path path/to/rag_store.pkl --query "what is diabetes"
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 / "rag_store.pkl"
8
  DEFAULT_MODEL = "BAAI/bge-large-en-v1.5"
9
- GOOGLE_DRIVE_FILE_ID = "1z3b0hd6-TFUu1fnrUi0gd1C0zugpSIv3"
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