Update app.py

app.py

@@ -79,8 +79,7 @@ elif task == "Sentiment Classification":
         
 
         elif input_text2 == "RNN":
-                with open("spam_model.pkl", 'rb') as model_file:
-                    rnn_model=pickle.load(model_file)
+                rnn_model=tf.keras.models.load_model("rnn_model.h5")
                 with open("spam_tokeniser.pkl", 'rb') as model_file:
                     rnn_tokeniser=pickle.load(model_file)
 
@@ -125,8 +124,7 @@ elif task == "Sentiment Classification":
                             
 
         elif input_text2 == "LSTM":
-                with open("lstm_model.pkl",'rb') as file:
-                    lstm_model = pickle.load(file) 
+                lstm_model=tf.keras.models.load_model("lstm_model.h5")
 
                 with open("lstm_tokeniser.pkl",'rb') as file:
                     lstm_tokeniser = pickle.load(file)