Update app.py

app.py

@@ -144,10 +144,11 @@ def generate_code_with_context(prompt, files, max_length=1024, temperature=0.7,
             return_attention_mask=True
         )
     
-        # Генерация с использованием обновленного кода
+        # Генерация с использованием обновленного кода и attention_mask
         with torch.no_grad():
             generated_ids = model.generate(
                 inputs.input_ids,
+                attention_mask=inputs.attention_mask,  # Добавляем attention_mask
                 max_new_tokens=max_length,
                 temperature=temperature,
                 top_p=top_p,
@@ -222,7 +223,7 @@ with gr.Blocks(title="Qwen2.5-Coder-14B with MCP", theme=gr.themes.Soft()) as de
             )
     
     # Добавляем информацию о MCP
-    with gr.Accordion("🔗 MCP Сервер - Подключение к другим приложениями", open=True):
+    with gr.Accordion("🔗 MCP Сервер - Подключение к другим приложениям", open=True):
         gr.Markdown("""
         **MCP (Model Context Protocol) сервер активирован!**