eeshanyaj commited on
Commit
42fe10b
·
1 Parent(s): 622abd8

policy model changed

Browse files
Files changed (2) hide show
  1. app/config.py +1 -1
  2. app/ml/policy_network.py +1 -1
app/config.py CHANGED
@@ -70,7 +70,7 @@ class Settings:
70
  # ========================================================================
71
  # MODEL PATHS (for RL Policy Network and RAG models)
72
  # ========================================================================
73
- POLICY_MODEL_PATH: str = os.getenv("POLICY_MODEL_PATH", "app/models/best_policy_model.pth")
74
  RETRIEVER_MODEL_PATH: str = os.getenv("RETRIEVER_MODEL_PATH", "app/models/best_retriever_model.pth")
75
  FAISS_INDEX_PATH: str = os.getenv("FAISS_INDEX_PATH", "app/models/faiss_index.pkl")
76
  KB_PATH: str = os.getenv("KB_PATH", "app/data/final_knowledge_base.jsonl")
 
70
  # ========================================================================
71
  # MODEL PATHS (for RL Policy Network and RAG models)
72
  # ========================================================================
73
+ POLICY_MODEL_PATH: str = os.getenv("POLICY_MODEL_PATH", "app/models/policy_query_only.pt")
74
  RETRIEVER_MODEL_PATH: str = os.getenv("RETRIEVER_MODEL_PATH", "app/models/best_retriever_model.pth")
75
  FAISS_INDEX_PATH: str = os.getenv("FAISS_INDEX_PATH", "app/models/faiss_index.pkl")
76
  KB_PATH: str = os.getenv("KB_PATH", "app/data/final_knowledge_base.jsonl")
app/ml/policy_network.py CHANGED
@@ -251,7 +251,7 @@ def load_policy_model() -> PolicyNetwork:
251
  if POLICY_MODEL is None:
252
  # Download model from HF Hub if needed (for deployment)
253
  settings.download_model_if_needed(
254
- hf_filename="models/best_policy_model.pth",
255
  local_path=settings.POLICY_MODEL_PATH
256
  )
257
 
 
251
  if POLICY_MODEL is None:
252
  # Download model from HF Hub if needed (for deployment)
253
  settings.download_model_if_needed(
254
+ hf_filename="models/policy_query_only.pt",
255
  local_path=settings.POLICY_MODEL_PATH
256
  )
257