deveix commited on
Commit
9d80a19
·
1 Parent(s): 902d012

change models

Browse files
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -140,8 +140,8 @@ async def get_answer(item: Item, token: str = Depends(verify_token)):
140
  raise HTTPException(status_code=500, detail=str(e))
141
 
142
  # mlp
143
- mlp_model = joblib.load('app/mlp_model2.pkl')
144
- mlp_pca = joblib.load('app/pca2.pkl')
145
  mlp_scaler = joblib.load('app/scaler.pkl')
146
  mlp_label_encoder = joblib.load('app/label_encoder.pkl')
147
 
 
140
  raise HTTPException(status_code=500, detail=str(e))
141
 
142
  # mlp
143
+ mlp_model = joblib.load('app/mlp_model.pkl')
144
+ mlp_pca = joblib.load('app/pca.pkl')
145
  mlp_scaler = joblib.load('app/scaler.pkl')
146
  mlp_label_encoder = joblib.load('app/label_encoder.pkl')
147