Rom89823974978 commited on
Commit
5ec54fa
·
1 Parent(s): 615ab66

updated source rag

Browse files
Files changed (1) hide show
  1. 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://mda_eu_project/data/consolidated_clean_pred.parquet"
60
- whoosh_dir: str = "gs://mda_eu_project/whoosh_index"
61
- vectorstore_path: str = "gs://mda_eu_project/vectorstore_index"
62
  # Models
63
  embedding_model: str = "sentence-transformers/LaBSE"
64
- llm_model: str = "google/flan-t5-base"
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