Gowthamvemula commited on
Commit
1b086b1
·
verified ·
1 Parent(s): 95da6ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,8 +3,9 @@ import pickle
3
  import numpy as np
4
 
5
  # Load model
6
- with open("log_reg_model.pkl", "rb") as f:
7
- model = pickle.load(f)
 
8
 
9
  # Define prediction function
10
  def predict_sleep(step: float, hour: float):
 
3
  import numpy as np
4
 
5
  # Load model
6
+ import joblib
7
+ model = joblib.load("log_reg_model.pkl")
8
+
9
 
10
  # Define prediction function
11
  def predict_sleep(step: float, hour: float):