ierhon commited on
Commit
d20c69f
1 Parent(s): ce0df62

fix cache ignore

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def train(data: str, message: str):
69
  model.compile(loss="categorical_crossentropy", metrics=["accuracy",])
70
 
71
  model.fit(X, y, epochs=64, batch_size=8, workers=4, use_multiprocessing=True)
72
- model.save("{data_hash}.keras")
73
  tokens = tokenizer.texts_to_sequences([message,])[0]
74
  prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
75
  max_o = 0
 
69
  model.compile(loss="categorical_crossentropy", metrics=["accuracy",])
70
 
71
  model.fit(X, y, epochs=64, batch_size=8, workers=4, use_multiprocessing=True)
72
+ model.save("cache/{data_hash}.keras")
73
  tokens = tokenizer.texts_to_sequences([message,])[0]
74
  prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
75
  max_o = 0