Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,9 @@ from keras.layers import Embedding, Dense, Dropout, Flatten, PReLU
|
|
5 |
from keras.preprocessing.text import Tokenizer
|
6 |
from keras_self_attention import SeqSelfAttention, SeqWeightedAttention
|
7 |
|
|
|
|
|
|
|
8 |
def train(data: str, message: str):
|
9 |
if "→" not in data and "\n" not in data:
|
10 |
return "Dataset should be like:\nquestion→answer\nquestion→answer\netc."
|
|
|
5 |
from keras.preprocessing.text import Tokenizer
|
6 |
from keras_self_attention import SeqSelfAttention, SeqWeightedAttention
|
7 |
|
8 |
+
emb_size = 128
|
9 |
+
inp_len = 16
|
10 |
+
|
11 |
def train(data: str, message: str):
|
12 |
if "→" not in data and "\n" not in data:
|
13 |
return "Dataset should be like:\nquestion→answer\nquestion→answer\netc."
|