Update app.py

app.py

@@ -18,6 +18,7 @@ import concurrent.futures
 import multiprocessing
 import io
 import os
+import tempfile
 
 import numpy as np
 from tensorflow.keras import Sequential
@@ -143,8 +144,11 @@ async def train_and_save_model():
 
         print("Loading or creating model...")
         if r.exists('chatbot_model'):
-            with io.BytesIO(r.get('chatbot_model')) as f:
-                model = load_model(f)
+            with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+                temp_file.write(r.get('chatbot_model'))
+                temp_file_name = temp_file.name
+            model = load_model(temp_file_name)
+            os.remove(temp_file_name)
         else:
             input_layer = Input(shape=(len(train_x[0]),))
             layer1 = Dense(128, activation='relu')(input_layer)
@@ -199,7 +203,11 @@ async def chat(message: ChatMessage):
     words = pickle.loads(r.get('words'))
     classes = pickle.loads(r.get('classes'))
     with io.BytesIO(r.get('chatbot_model')) as f:
-        model = load_model(f)
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            temp_file.write(f.read())
+            temp_file_name = temp_file.name
+        model = load_model(temp_file_name)
+        os.remove(temp_file_name)
 
     sentence_words = nltk.word_tokenize(message.message)
     sentence_words = [lemmatizer.lemmatize(word.lower()) for word in sentence_words]
@@ -220,8 +228,7 @@ async def chat(message: ChatMessage):
 
     r.rpush('user_questions', message.message)
 
-    # Auto-train after receiving a new message
-    asyncio.create_task(train_and_save_model()) 
+    asyncio.create_task(train_and_save_model())
 
     return return_list