Update main.py

main.py

@@ -121,12 +121,12 @@ def API_France_Travail_Metier(metier):
 def Connexion_Mistral():
     #endpoint = "https://models.inference.ai.azure.com"
     #return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
-    repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
-    #repo_id = "Qwen/Qwen2.5-3B-Instruct"
+    #repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
+    repo_id = "Qwen/Qwen2.5-3B-Instruct"
     #repo_id = "microsoft/Phi-3.5-mini-instruct"
     #repo_id = "meta-llama/Llama-3.2-3B-Instruct"
     llm = HuggingFaceEndpoint(
-        repo_id=repo_id, max_new_tokens=5000, temperature=0.1, task="text2text-generation", streaming=True
+        repo_id=repo_id, max_new_tokens=1024, temperature=0.1, task="text2text-generation", streaming=True
     )
     return llm