Update pages/gpt.py

pages/gpt.py

@@ -1,5 +1,5 @@
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
-
+import torch
 
 model = GPT2LMHeadModel.from_pretrained(
     'sberbank-ai/rugpt3small_based_on_gpt2',
@@ -10,8 +10,6 @@ model = GPT2LMHeadModel.from_pretrained(
 model.load_state_dict(torch.load('model.pt'))
 
 
-import torch
-
 def generate_text(model, tokenizer, prompt, length, num_samples, temperature):
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     output_sequences = model.generate(