Update app.py

app.py

@@ -27,7 +27,7 @@ def load_custom_model():
 # Функция для генерации текста
 def generate_text(model, tokenizer, prompt, max_len, temperature):
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
-    attention_mask = (input_ids != tokenizer.pad_token_id).to(dtype=torch.long)
+    attention_mask = torch.ones_like(input_ids)
     # Генерация текста
     output = model.generate(
         input_ids,