Spaces:

sehaj13
/

Text_vision_mixing_prediction

Sleeping

sehaj13 commited on Jun 1

Commit

d6f04ba

verified ·

1 Parent(s): 6506252

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import joblib
 # Load model and preprocessing objects
 model = joblib.load("mixing_prediction_model.pkl")
 scaler = joblib.load("scaler.pkl")
-label_encoder = joblib.load("label_encoder.pkl")
 # Define prediction function
 def predict_class(COUNT, COUNT_CV, STRENGTH, CSP, U_PERCENT, THIN, THICK, NEPS, IPI):

 # Load model and preprocessing objects
 model = joblib.load("mixing_prediction_model.pkl")
 scaler = joblib.load("scaler.pkl")
+label_encoder = joblib.load("label_encoder_new.pkl")
 # Define prediction function
 def predict_class(COUNT, COUNT_CV, STRENGTH, CSP, U_PERCENT, THIN, THICK, NEPS, IPI):