Update app.py

app.py

@@ -6,8 +6,8 @@ import torch.nn.functional as F
 
 from transformers import AutoTokenizer, GPT2LMHeadModel
 tokenizer = AutoTokenizer.from_pretrained("supermy/jinyong-gpt2")
-tokenizer.add_special_tokens(['SEP'])
-tokenizer.add_special_tokens(['UNK'])
+tokenizer.add_special_tokens({'sep_token': '[SEP]'})
+tokenizer.add_special_tokens({'unk_token': '[UNK]'})
 model = GPT2LMHeadModel.from_pretrained("supermy/jinyong-gpt2")
 model.eval()