Spaces:
Sleeping
Sleeping
policy model changed
Browse files- app/config.py +1 -1
- 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/
|
| 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/
|
| 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 |
|