Spaces:
Sleeping
Sleeping
fix cache ignore
Browse files
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
|