Update main.py

main.py

@@ -122,11 +122,15 @@ 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 = "Qwen/Qwen2.5-3B-Instruct"
+    repo_id = "HuggingFaceH4/zephyr-7b-beta"
+    #repo_id = "HuggingFaceH4/zephyr-7b-alpha"
+    #repo_id = "tiiuae/falcon-7b-instruct"
+    #repo_id = "HuggingFaceTB/SmolLM2-1.7B-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=1024, temperature=0.1, task="text2text-generation", streaming=True
+        repo_id=repo_id, max_new_tokens=3000, temperature=0.1, task="text2text-generation", streaming=True
     )
     return llm