Update app.py
Browse files
app.py
CHANGED
@@ -189,7 +189,6 @@ elif selected_model == "Генерация текста GPT-моделью по
|
|
189 |
model_name_or_path = "sberbank-ai/rugpt3small_based_on_gpt2"
|
190 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
|
191 |
model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
|
192 |
-
model = GPT2LMHeadModel(config)
|
193 |
model.load_state_dict(torch.load('model_dict.pt', map_location=device))
|
194 |
model.eval()
|
195 |
|
|
|
189 |
model_name_or_path = "sberbank-ai/rugpt3small_based_on_gpt2"
|
190 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
|
191 |
model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
|
|
|
192 |
model.load_state_dict(torch.load('model_dict.pt', map_location=device))
|
193 |
model.eval()
|
194 |
|