ElijahDi commited on
Commit
0a3c50b
·
verified ·
1 Parent(s): 8141906

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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