Update app.py

app.py

@@ -7,6 +7,7 @@ from langchain import hub
 from langchain.agents import AgentExecutor, create_react_agent
 from langchain_core.prompts import PromptTemplate
 from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_community.llms import HuggingFaceHub
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HF_TOKEN")
@@ -41,9 +42,9 @@ search_tool = DuckDuckGoSearchRun()
 tools = [search_tool]
 
 # LLM
-model = HuggingFaceEndpoint(
-    repo_id="google/t5gemma-b-b-prefixlm",
-    huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+llm = HuggingFaceHub(
+    repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+    model_kwargs={"temperature": 0.1}
 )
 
 class BasicAgent: