Zakia commited on
Commit
8a1db3f
1 Parent(s): b90b648

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -2,12 +2,11 @@
2
  # Almadadh Ya Gause Radi Allahu Ta'alah Anh - Ameen
3
 
4
 
5
- import pickle
6
  import gradio as gr
7
 
8
- # Load the trained model from .pkl file
9
- with open('iris_decision_tree_model.pkl', 'rb') as file:
10
- clf = pickle.load(file)
11
 
12
  # Import iris dataset for target names
13
  from sklearn import datasets
 
2
  # Almadadh Ya Gause Radi Allahu Ta'alah Anh - Ameen
3
 
4
 
5
+ from joblib import load
6
  import gradio as gr
7
 
8
+ # Load the trained model
9
+ clf = load('iris_decision_tree_model.joblib')
 
10
 
11
  # Import iris dataset for target names
12
  from sklearn import datasets