Update app.py

app.py

@@ -188,7 +188,7 @@ elif selected_model == "Генерация текста GPT-моделью по
     # GPT2
     model_name_or_path = "sberbank-ai/rugpt3small_based_on_gpt2"
     tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
-    model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(DEVICE)
+    model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
     model = GPT2LMHeadModel(config)
     model.load_state_dict(torch.load('model_dict.pt', map_location=device))
     model.eval()