ajeetkumar01 commited on
Commit
51b5122
·
verified ·
1 Parent(s): 063f4b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import gradio as gr
3
  import pandas as pd
4
 
5
  # Load the preprocessor and the best model
6
- preprocessor = joblib.load('/preprocessor.pkl')
7
- best_model = joblib.load('/best_model_gradient_boosting.pkl')
8
 
9
  # Define the prediction function
10
  def predict_income(age, workclass, fnlwgt, education, education_num, marital_status, occupation, relationship, race, sex, capital_gain, capital_loss, hours_per_week, native_country):
 
3
  import pandas as pd
4
 
5
  # Load the preprocessor and the best model
6
+ preprocessor = joblib.load('preprocessor.pkl')
7
+ best_model = joblib.load('best_model_gradient_boosting.pkl')
8
 
9
  # Define the prediction function
10
  def predict_income(age, workclass, fnlwgt, education, education_num, marital_status, occupation, relationship, race, sex, capital_gain, capital_loss, hours_per_week, native_country):