Update app.py

app.py

@@ -4,8 +4,9 @@ import torch
 import gradio as gr
 import torch.nn.functional as F
 
-from transformers import BertTokenizer, GPT2LMHeadModel
-tokenizer = BertTokenizer.from_pretrained("supermy/poetry")
+from transformers import BertTokenizer, GPT2LMHeadModel,PreTrainedTokenizerFast
+# tokenizer = BertTokenizer.from_pretrained("supermy/poetry")
+tokenizer = BertTokenizer(vocab_file="poetry-bpe.json")
 model = GPT2LMHeadModel.from_pretrained("supermy/poetry")
 model.eval()