MoinulwithAI commited on
Commit
0ee9995
·
verified ·
1 Parent(s): 2aafbf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from pytorch_tabnet.tab_model import TabNetClassifier
5
 
6
  # Load model, scaler, and encoder
7
  model = TabNetClassifier()
8
- model.load_model('tabnet_model')
9
 
10
  scaler = joblib.load('D:/Dataset/IPIP-FFM-data-8Nov2018/scaler.save')
11
  encoder = joblib.load('D:/Dataset/IPIP-FFM-data-8Nov2018/encoder.save')
 
5
 
6
  # Load model, scaler, and encoder
7
  model = TabNetClassifier()
8
+ model.load_model('tabnet_model.zip')
9
 
10
  scaler = joblib.load('D:/Dataset/IPIP-FFM-data-8Nov2018/scaler.save')
11
  encoder = joblib.load('D:/Dataset/IPIP-FFM-data-8Nov2018/encoder.save')