DaniloTertu commited on
Commit
50488f8
·
1 Parent(s): b185204

Upload 2 files

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. app.py +2 -2
requirements.txt CHANGED
@@ -1 +1,2 @@
1
  scikit-learn
 
 
1
  scikit-learn
2
+ joblib
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
- import pandas as pd
3
 
4
- m = pd.read_pickle('tree_model.pkl')
5
 
6
  def predict(gender, ever_married, work_type, residence_type, smoking_status, age, hypertension, heart_disease, avg_glucose_level, bmi):
7
  input_data = {
 
1
  import gradio as gr
2
+ import joblib
3
 
4
+ m = joblib.load('tree_model.pkl')
5
 
6
  def predict(gender, ever_married, work_type, residence_type, smoking_status, age, hypertension, heart_disease, avg_glucose_level, bmi):
7
  input_data = {