Update main.py

main.py

@@ -19,13 +19,13 @@ def load_model(model_path):
     return model
 
 
-def load_tokenizer(tokenizer_path):
-    tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
+def load_tokenizer(checkpoint):
+    tokenizer = GPT2Tokenizer.from_pretrained(checkpoint)
     return tokenizer
 
 model_path = r'./checkpoint/'
 model = load_model(model_path)
-tokenizer = load_tokenizer(model_path)
+tokenizer = load_tokenizer('gpt2')
 
 
 def generate_text(sequence, max_new_tokens):