Update app.py

app.py

@@ -1,18 +1,27 @@
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+
+# Load the pre-trained model and tokenizer
 tokenizer = GPT2Tokenizer.from_pretrained("sberbank-ai/mGPT")
 model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT")
 
+# Input text
 text = "Александр Сергеевич Пушкин родился в "
-input_ids = tokenizer.encode(text, return_tensors="pt").cuda(device)
+
+# Encode the input text
+input_ids = tokenizer.encode(text, return_tensors="pt")
+
+# Generate text
 out = model.generate(
-        input_ids, 
-        min_length=100, 
-        max_length=100, 
-        eos_token_id=5, 
-        pad_token=1,
-        top_k=10,
-        top_p=0.0,
-        no_repeat_ngram_size=5
+    input_ids,
+    min_length=100,
+    max_length=100,
+    eos_token_id=5,
+    pad_token_id=1,
+    top_k=10,
+    top_p=0.0,
+    no_repeat_ngram_size=5
 )
+
+# Decode the generated output
 generated_text = list(map(tokenizer.decode, out))[0]
-print(generated_text)
+print(generated_text)