Update app.py

app.py

@@ -150,20 +150,20 @@ col_change = load(dst_path)
 # quality change
 src_path = hf_hub_download(
     repo_id="WebashalarForML/Diamond_model_",
-    filename="CLASS_DUMMY/LR_best__2_class_qua_change.pkl",
+    filename="CLASS_DUMMY/DT_best__2_class_qua_change.pkl",
     cache_dir=MODEL_FOLDER
 )
-dst_path = os.path.join(MODEL_FOLDER, "LR_best__2_class_qua_change.pkl")
+dst_path = os.path.join(MODEL_FOLDER, "DT_best__2_class_qua_change.pkl")
 shutil.copy(src_path, dst_path)
 qua_change = load(dst_path)
 
 # cut change
 src_path = hf_hub_download(
     repo_id="WebashalarForML/Diamond_model_",
-    filename="CLASS_DUMMY/LR_best__2_class_cut_change.pkl",
+    filename="CLASS_DUMMY/DT_best__2_class_cut_change.pkl",
     cache_dir=MODEL_FOLDER
 )
-dst_path = os.path.join(MODEL_FOLDER, "LR_best__2_class_cut_change.pkl")
+dst_path = os.path.join(MODEL_FOLDER, "DT_best__2_class_cut_change.pkl")
 shutil.copy(src_path, dst_path)
 cut_change = load(dst_path)