Mpavan45 commited on
Commit
b23de27
·
verified ·
1 Parent(s): d4c35a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -217,9 +217,7 @@ def load_model():
217
  # Load the model
218
  model = tf.keras.models.load_model("model_m3_new.keras")
219
 
220
- # Load vectorizer (use pickle or joblib for sklearn models)
221
- with open("vec_text_m3_new.pkl", 'rb') as file:
222
- vectorizer = pickle.load(file)
223
 
224
  # Load label encoder
225
  with open("label_encoder_m5.pkl", 'rb') as file:
 
217
  # Load the model
218
  model = tf.keras.models.load_model("model_m3_new.keras")
219
 
220
+ vectorizer = keras.models.load_model("vec_text_m3_new.keras")
 
 
221
 
222
  # Load label encoder
223
  with open("label_encoder_m5.pkl", 'rb') as file: