Update README.md

README.md

@@ -22,9 +22,10 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import torch
 DEVICE = torch.device("cuda:0")
 
-model_name_or_path = "sberbank-ai/rugpt3medium_based_on_gpt2"
-tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
-model = GPT2LMHeadModel.from_pretrained('model').to(DEVICE)
+model_name_or_path = "radm/rugpt3medium-tathagata"
+tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/rugpt3medium_based_on_gpt2")
+model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(DEVICE)
+
 text = "В чем смысл жизни?\n"
 input_ids = tokenizer.encode(text, return_tensors="pt").to(DEVICE)
 model.eval()