Spaces:
Sleeping
Sleeping
Commit
·
5ec54fa
1
Parent(s):
615ab66
updated source rag
Browse files- backend/rag.py +4 -4
backend/rag.py
CHANGED
@@ -56,12 +56,12 @@ logger = logging.getLogger(__name__)
|
|
56 |
|
57 |
class Settings(BaseSettings):
|
58 |
# Parquet + Whoosh/FAISS
|
59 |
-
parquet_path: str = "gs://
|
60 |
-
whoosh_dir: str = "gs://
|
61 |
-
vectorstore_path: str = "gs://
|
62 |
# Models
|
63 |
embedding_model: str = "sentence-transformers/LaBSE"
|
64 |
-
llm_model: str = "google/
|
65 |
cross_encoder_model: str = "cross-encoder/mmarco-mMiniLMv2-L12-H384-v1"
|
66 |
# RAG parameters
|
67 |
chunk_size: int = 750
|
|
|
56 |
|
57 |
class Settings(BaseSettings):
|
58 |
# Parquet + Whoosh/FAISS
|
59 |
+
parquet_path: str = "gs://mda_kul_project/data/consolidated_clean_pred.parquet"
|
60 |
+
whoosh_dir: str = "gs://mda_kul_project/whoosh_index"
|
61 |
+
vectorstore_path: str = "gs://mda_kul_project/vectorstore_index"
|
62 |
# Models
|
63 |
embedding_model: str = "sentence-transformers/LaBSE"
|
64 |
+
llm_model: str = "google/flmt5-base"
|
65 |
cross_encoder_model: str = "cross-encoder/mmarco-mMiniLMv2-L12-H384-v1"
|
66 |
# RAG parameters
|
67 |
chunk_size: int = 750
|