luciancotolan commited on
Commit
cd62a49
·
1 Parent(s): 98a9e45
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import gradio as gr
2
  import pandas as pd
3
- import joblib
4
 
5
- treemodel = joblib.load('decision_tree.pkl')
6
- nnmodel = joblib.load('nnmodel.h5')
7
 
8
  def onehot(df, column):
9
  df = df.copy()
 
1
  import gradio as gr
2
  import pandas as pd
3
+ import pickle
4
 
5
+ treemodel = pickle.load('decision_tree.pkl')
6
+ nnmodel = pickle.load('neural_network.pkl')
7
 
8
  def onehot(df, column):
9
  df = df.copy()
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- joblib
2
  sklearn
3
  scikit-learn
4
  keras
 
1
+ pickle
2
  sklearn
3
  scikit-learn
4
  keras