Spaces:
Runtime error
Runtime error
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
|