Update app.py

app.py

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