Update app.py
Browse files
app.py
CHANGED
|
@@ -81,10 +81,10 @@ os.makedirs(app.config['MODEL_FOLDER'], exist_ok=True)
|
|
| 81 |
# Prediction analysis models loaded from Hugging Face.
|
| 82 |
src_path = hf_hub_download(
|
| 83 |
repo_id="WebashalarForML/Diamond_model_",
|
| 84 |
-
filename="models_list/mkble/DecisionTree_best_pipeline_mkble_0_to_0.
|
| 85 |
cache_dir=MODEL_FOLDER
|
| 86 |
)
|
| 87 |
-
dst_path = os.path.join(MODEL_FOLDER, "DecisionTree_best_pipeline_mkble_0_to_0.
|
| 88 |
shutil.copy(src_path, dst_path)
|
| 89 |
makable_model = load(dst_path)
|
| 90 |
|
|
|
|
| 81 |
# Prediction analysis models loaded from Hugging Face.
|
| 82 |
src_path = hf_hub_download(
|
| 83 |
repo_id="WebashalarForML/Diamond_model_",
|
| 84 |
+
filename="models_list/mkble/DecisionTree_best_pipeline_mkble_0_to_0.99_al.pkl",
|
| 85 |
cache_dir=MODEL_FOLDER
|
| 86 |
)
|
| 87 |
+
dst_path = os.path.join(MODEL_FOLDER, "DecisionTree_best_pipeline_mkble_0_to_0.99_al.pkl")
|
| 88 |
shutil.copy(src_path, dst_path)
|
| 89 |
makable_model = load(dst_path)
|
| 90 |
|