Spaces:
Sleeping
Sleeping
Commit Β·
0314823
1
Parent(s): 81726c9
updated top_k value
Browse files- rag/retriever.py +1 -1
rag/retriever.py
CHANGED
|
@@ -12,7 +12,7 @@ load_dotenv()
|
|
| 12 |
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
|
| 13 |
PINECONE_INDEX = os.getenv("PINECONE_INDEX", "study-saathi")
|
| 14 |
EMBEDDING_MODEL = "intfloat/multilingual-e5-large"
|
| 15 |
-
TOP_K =
|
| 16 |
|
| 17 |
# ββ Device βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 18 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 12 |
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
|
| 13 |
PINECONE_INDEX = os.getenv("PINECONE_INDEX", "study-saathi")
|
| 14 |
EMBEDDING_MODEL = "intfloat/multilingual-e5-large"
|
| 15 |
+
TOP_K = 10
|
| 16 |
|
| 17 |
# ββ Device βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 18 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|