Update agent.py

agent.py

@@ -458,8 +458,6 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
 #llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
 
 
-
-# Get the Hugging Face API token from the environment variable
 # Get the Hugging Face API token from the environment variable
 hf_token = os.getenv("HF_TOKEN")
 
@@ -468,13 +466,13 @@ from langchain_huggingface import HuggingFaceEndpoint
 
 # Initialize the HuggingFaceEndpoint with the desired model and parameters
 llm = HuggingFaceEndpoint(
-    repo_id="code_llama/code_llama-7b-instruct",
+    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     task="text-generation",
-    huggingfacehub_api_token=os.getenv("HF_TOKEN"),
-    temperature=0.7,
-    max_new_tokens=1024
+    huggingfacehub_api_token=hf_token,
+    model_kwargs={"temperature": 0.7, "max_new_tokens": 1024}
 )
 
+
 # Initialize the LangChain agent with the tool(s) and the model
 agent = initialize_agent(
     tools=tools,