Update app.py
Browse files
app.py
CHANGED
@@ -57,9 +57,9 @@ def generate(title, context, max_len):
|
|
57 |
|
58 |
if __name__ == '__main__':
|
59 |
# tokenizer = BertTokenizer(vocab_file="chinese_vocab.model")
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
|
64 |
|
65 |
|
|
|
57 |
|
58 |
if __name__ == '__main__':
|
59 |
# tokenizer = BertTokenizer(vocab_file="chinese_vocab.model")
|
60 |
+
eod_id = tokenizer.convert_tokens_to_ids("<eod>")
|
61 |
+
sep_id = tokenizer.sep_token_id
|
62 |
+
unk_id = tokenizer.unk_token_id
|
63 |
|
64 |
|
65 |
|