WebashalarForML commited on
Commit
d4b3c99
·
verified ·
1 Parent(s): 577cf9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -9
app.py CHANGED
@@ -45,19 +45,46 @@ os.makedirs(app.config['MODEL_FOLDER'], exist_ok=True)
45
 
46
  # prediction analysis
47
  # Download the model file to the specified location
48
- file_path = hf_hub_download(
49
  repo_id="WebashalarForML/Diamond_model_",
50
- filename="models_list/bygrad/CatBoost_best_pipeline_BYGRADE.pkl",
51
- cache_dir=specific_location
52
  )
53
 
54
- with open(file_path, "rb") as f:
55
- model = pickle.load(f)
56
 
57
- gia_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_gia_price.joblib'))
58
- grade_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_grade_price.joblib'))
59
- bygrade_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_bygrade_price.joblib'))
60
- makable_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_makable_price.joblib'))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
61
 
62
  # classifcation analysis
63
  col_model = load(os.path.join(MODEL_DIR, 'classification_LogisticRegression_col.joblib'))
 
45
 
46
  # prediction analysis
47
  # Download the model file to the specified location
48
+ file_path_1 = hf_hub_download(
49
  repo_id="WebashalarForML/Diamond_model_",
50
+ filename="models_list/mkble/StackingRegressor_best_pipeline_mkble_0_to_1.01.pkl",
51
+ cache_dir=MODEL_FOLDER
52
  )
53
 
54
+ with open(file_path_1, "rb") as f:
55
+ makable_model = pickle.load(f)
56
 
57
+ file_path_2 = hf_hub_download(
58
+ repo_id="WebashalarForML/Diamond_model_",
59
+ filename="models_list/grd/StackingRegressor_best_pipeline_grd_0_to_1.01.pkl",
60
+ cache_dir=MODEL_FOLDER
61
+ )
62
+
63
+ with open(file_path_2, "rb") as f:
64
+ grade_model = pickle.load(f)
65
+
66
+ file_path_3 = hf_hub_download(
67
+ repo_id="WebashalarForML/Diamond_model_",
68
+ filename="models_list/bygrad/StackingRegressor_best_pipeline_bygrad_0_to_1.01.pkl",
69
+ cache_dir=MODEL_FOLDER
70
+ )
71
+
72
+ with open(file_path_3, "rb") as f:
73
+ bygrade_model = pickle.load(f)
74
+
75
+ file_path_4 = hf_hub_download(
76
+ repo_id="WebashalarForML/Diamond_model_",
77
+ filename="models_list/gia/StackingRegressor_best_pipeline_gia_0_to_1.01.pkl",
78
+ cache_dir=MODEL_FOLDER
79
+ )
80
+
81
+ with open(file_path_4, "rb") as f:
82
+ gia_model = pickle.load(f)
83
+
84
+ #gia_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_gia_price.joblib'))
85
+ #grade_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_grade_price.joblib'))
86
+ #bygrade_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_bygrade_price.joblib'))
87
+ #makable_model = load(os.path.join(MODEL_DIR, 'linear_regression_model_makable_price.joblib'))
88
 
89
  # classifcation analysis
90
  col_model = load(os.path.join(MODEL_DIR, 'classification_LogisticRegression_col.joblib'))