Fix cache not found & memory leak

app.py

@@ -7,6 +7,7 @@ from keras.layers import *
 from keras.preprocessing.text import Tokenizer
 import os
 import hashlib
+import keras
 
 os.mkdir("cache")
 
@@ -28,7 +29,7 @@ def train(data: str, message: str):
     vocab_size = len(tokenizer.word_index) + 1
     data_hash = hash_str(data)+".keras"
     if data_hash in os.listdir("cache"):
-        model = load_model(data_hash)
+        model = load_model("cache/"+data_hash)
     else:
         input_layer = Input(shape=(inp_len,))
         emb_layer = Embedding(input_dim=vocab_size, output_dim=emb_size, input_length=inp_len)(input_layer)
@@ -77,6 +78,7 @@ def train(data: str, message: str):
         if max_v < i:
             max_v = i
             max_o = ind
+    keras.backend.clear_session()
     return responses[ind]
 
 iface = gr.Interface(fn=train, inputs=["text", "text"], outputs="text")