Update app.py

app.py

@@ -18,7 +18,7 @@ import numpy as np
 from tensorflow.keras import Sequential, Input
 from tensorflow.keras.layers import Dense, Dropout
 from tensorflow.keras.optimizers import SGD
-from tensorflow.keras.models import load_model
+from tensorflow.keras.models import load_model, save_model
 
 import redis
 import os
@@ -132,8 +132,8 @@ async def train_and_save_model():
         train_y = list(training[:, 1])
 
         print("Loading or creating model...")
-        if r.exists('model'):
-            model = load_model('chatbot_model')
+        if os.path.exists('chatbot_model.h5'):
+            model = load_model('chatbot_model.h5')
         else:
             model = Sequential()
             model.add(Input(shape=(len(train_x[0]),)))
@@ -163,7 +163,7 @@ async def train_and_save_model():
             classes = sorted(list(set(existing_classes + classes)))
             r.set('classes', pickle.dumps(classes))
 
-        model.save('chatbot_model')
+        save_model(model, 'chatbot_model.h5')
 
         print("Data and model saved. Re-training...")
 
@@ -179,7 +179,7 @@ class ChatMessage(BaseModel):
 async def chat(message: ChatMessage):
     words = pickle.loads(r.get('words'))
     classes = pickle.loads(r.get('classes'))
-    model = load_model('chatbot_model')
+    model = load_model('chatbot_model.h5')
 
     sentence_words = nltk.word_tokenize(message.message)
     sentence_words = [lemmatizer.lemmatize(word.lower()) for word in sentence_words]