Update README.md

README.md

@@ -51,7 +51,7 @@ special_tokens = {
 }
 tokenizer.add_special_tokens(special_tokens)
 model.resize_token_embeddings(len(tokenizer))
-model.load_state_dict(torch.load("path/to/model))
+model.load_state_dict(torch.load("path/to/model"))
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 def generate_text(model, tokenizer, prompt, max_length=1024):