supermy commited on
Commit
a02deee
1 Parent(s): 3e475a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # 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
 
 
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