Spaces:
Sleeping
Sleeping
Commit
·
004784e
1
Parent(s):
6a8c4a1
1.1 init
Browse files
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/
|
| 48 |
-
"random_forest": "classifier/
|
| 49 |
-
"xgboost": "classifier/
|
| 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 |
# -------------------------------------------------------------------
|