SuriRaja commited on
Commit
31ad924
·
verified ·
1 Parent(s): bdf9b37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -37,8 +37,11 @@ def train_model(output_range):
37
  model = LinearRegression().fit(X, y)
38
  return model
39
 
 
 
 
40
  models = {
41
- "Hemoglobin": train_model((13.5, 17.5)),
42
  "WBC Count": train_model((4.0, 11.0)),
43
  "Platelet Count": train_model((150, 450)),
44
  "Iron": train_model((60, 170)),
 
37
  model = LinearRegression().fit(X, y)
38
  return model
39
 
40
+ import joblib
41
+ hemoglobin_model = joblib.load("hemoglobin_model.pkl")
42
+
43
  models = {
44
+ "Hemoglobin": hemoglobin_model,
45
  "WBC Count": train_model((4.0, 11.0)),
46
  "Platelet Count": train_model((150, 450)),
47
  "Iron": train_model((60, 170)),