Update app.py

app.py

@@ -89,8 +89,10 @@ if task=='Sentiment Classification':
 
     elif arc == arcs[2]:
         # DNN
-        with open("dnn_model.pkl",'rb') as file:
-            dnn_model = pickle.load(file)
+        dnn_model = load_model("dnn_model1.h5")
+
+        #with open("dnn_model.pkl",'rb') as file:
+            #dnn_model = pickle.load(file)
         with open("dnn_tokeniser.pkl",'rb') as file:
             dnn_tokeniser = pickle.load(file)
 
@@ -112,8 +114,9 @@ if task=='Sentiment Classification':
 
     elif arc == arcs[3]:
         # RNN 
-        with open("rnn_model.pkl",'rb') as file:
-            rnn_model = pickle.load(file) 
+        #with open("rnn_model.pkl",'rb') as file:
+            #rnn_model = pickle.load(file) 
+        rnn_model = load_model("rnn_model1.h5")
 
         with open("rnn_tokeniser.pkl",'rb') as file:
             rnn_tokeniser = pickle.load(file)
@@ -136,8 +139,9 @@ if task=='Sentiment Classification':
 
     elif arc == arcs[4]:
         # LSTM
-        with open("lstm_model.pkl",'rb') as file:
-            lstm_model = pickle.load(file) 
+        #with open("lstm_model.pkl",'rb') as file:
+            #lstm_model = pickle.load(file) 
+        lstm_model = load_model("lstm_model1.h5")
 
         with open("lstm_tokeniser.pkl",'rb') as file:
             lstm_tokeniser = pickle.load(file)