Update app.py

app.py

@@ -14,6 +14,7 @@ from bs4 import BeautifulSoup
 from concurrent.futures import ThreadPoolExecutor
 import re
 
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = expandable_segments:True
 # --- Configuration du logger ---
 logging.basicConfig(
     level=logging.INFO,
@@ -28,6 +29,7 @@ logging.basicConfig(
 # Assurez-vous que la variable d'environnement HF_TOKEN est définie avec votre token Hugging Face
 # Sinon, vous pouvez la définir directement ici
 # os.environ["HF_TOKEN"] = "votre_token_huggingface"
+
 login(token=os.environ["HF_TOKEN"])
 
 # === Chargement des modèles ===
@@ -59,7 +61,7 @@ analyzer_model = AutoModelForCausalLM.from_pretrained(
 analyzer_tokenizer = AutoTokenizer.from_pretrained(analyzer_model_name)
 
 # AgentCoder
-coder_model_name = "Qwen/Qwen2.5-Coder-14B-Instruct"
+coder_model_name = "Qwen/Qwen2.5-Coder-7B-Instruct"
 coder_model = AutoModelForCausalLM.from_pretrained(
     coder_model_name,
     torch_dtype="auto",