NeerajCodz commited on
Commit
004784e
·
1 Parent(s): 6a8c4a1
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -44,9 +44,9 @@ MODELS = {} # Global dictionary to store loaded pipelines
44
 
45
  # Update this map based on the actual model names saved by your training script
46
  MODEL_MAP = {
47
- "decision_tree": "classifier/ccfd_1.0_decision-tree.pkl",
48
- "random_forest": "classifier/ccfd_1.0_random-forest.pkl",
49
- "xgboost": "classifier/ccfd_1.0_xg-boost.pkl",
50
  }
51
 
52
  # -------------------------------------------------------------------
 
44
 
45
  # Update this map based on the actual model names saved by your training script
46
  MODEL_MAP = {
47
+ "decision_tree": f"classifier/ccfd_{VERSION}_decision-tree.pkl",
48
+ "random_forest": f"classifier/ccfd_{VERSION}_random-forest.pkl",
49
+ "xgboost": f"classifier/ccfd_{VERSION}_xg-boost.pkl",
50
  }
51
 
52
  # -------------------------------------------------------------------