Hugging Face's logo

Spaces:
princeml
/
Health_insurance_premium_prediction
Runtime error

App Files Community
princeml commited on
Commit
8facc60
•
1 Parent(s): df0d049

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,10 +1,12 @@
1
  import streamlit as st
2
  import pandas as pd
3
  import numpy as np
4
- import pickle
 
5
 
6
  # load the model from disk
7
- loaded_model = pickle.load(open("random_forest_reg.sav", 'rb'))
 
8
 
9
 
10
  def generate_prediction(input_array):
 
1
  import streamlit as st
2
  import pandas as pd
3
  import numpy as np
4
+ import joblib
5
+
6
 
7
  # load the model from disk
8
+ filename = 'finalized_model.sav'
9
+ loaded_model = joblib.load(filename)
10
 
11
 
12
  def generate_prediction(input_array):