Spaces:

ierhon
/

neural-chatbot-constructor

Sleeping

ierhon commited on Feb 4

Commit

147d2ba

•

1 Parent(s): 228e5ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_le
     tokenizer.fit_on_texts(list(dset.keys()))
     vocab_size = len(tokenizer.word_index) + 1
-    data_hash = hash_str(data)+".keras"
     if data_hash in os.listdir("cache"):
         model = load_model("cache/"+data_hash)
     else:

     tokenizer.fit_on_texts(list(dset.keys()))
     vocab_size = len(tokenizer.word_index) + 1
+    data_hash = hash_str(data)+str(epochs)+str(learning_rate)+str(emb_size)+str(inp_len)+".keras"
     if data_hash in os.listdir("cache"):
         model = load_model("cache/"+data_hash)
     else: