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).long()
+    attention_mask = (input_ids != tokenizer.pad_token_id).to(dtype=torch.long)
     # Генерация текста
     output = model.generate(
         input_ids,
@@ -51,7 +51,7 @@ def main():
     model_gpt2, tokenizer_gpt2 = load_gpt2()  # GPT-2 модель
     model_custom, tokenizer_custom = load_custom_model()  # Кастомная модель
 
-    st.write("Fine-tuned GPT-2 for New Language with Custom Tokenizer")
+    #st.write("Fine-tuned GPT-2 for New Language with Custom Tokenizer")
 
     # # Блок для генерации текста с GPT-2
     # st.subheader("GPT-2 Text Generation")