Update app.py

app.py

@@ -35,9 +35,9 @@ def pre_process(text):
 
 @st.cache_resource
 def load_model():
-    model = tf.keras.models.load_model("model_m3_new.keras")
-    vectorizer = tf.keras.models.load_model("vec_text_m3_new.keras")
-    with open("label_encoder_m5.pkl", 'rb') as file:
+    model = tf.keras.models.load_model("news_model.keras")
+    vectorizer = tf.keras.models.load_model("news_tv_model.keras")
+    with open("label_encoder.pkl", 'rb') as file:
         label_encoder = pickle.load(file)
     return model, vectorizer, label_encoder
 
@@ -62,4 +62,4 @@ if st.button("Predict Category"):
         category = predict_category(user_text)
         st.success(f"Predicted Category: {category}")
     else:
-        st.warning("Please enter some text to classify.")
+        st.warning("Please enter some text to classify.")