Muzmmillcoste commited on
Commit
bbf29a4
·
1 Parent(s): 6ab38ef
Files changed (2) hide show
  1. app.py +3 -3
  2. logistic_model_pipeline.pkl2 +0 -0
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
- import joblib
3
  import pandas as pd
4
 
5
  # Load the saved model
6
- joblib_file = "logistic_model_pipeline.pkl"
7
- loaded_pipeline = joblib.load(joblib_file)
8
 
9
  # Define a function to make predictions
10
  def predict_heart_disease(age, sex_male, cigs_per_day, tot_chol, sys_bp, glucose):
 
1
  import gradio as gr
2
+ import pickle
3
  import pandas as pd
4
 
5
  # Load the saved model
6
+ with open("logistic_model_pipeline.pkl2", "rb") as f:
7
+ pipeline = pickle.load(f)
8
 
9
  # Define a function to make predictions
10
  def predict_heart_disease(age, sex_male, cigs_per_day, tot_chol, sys_bp, glucose):
logistic_model_pipeline.pkl2 ADDED
Binary file (1.41 kB). View file