Update app.py

app.py

@@ -34,6 +34,7 @@ from langchain.agents import initialize_agent, Tool
 from langchain_community.llms import HuggingFaceHub
 #from langchain_community.tools import PythonREPL
 
+from langchain_huggingface import HuggingFaceEndpoint
 #from langchain_community.tools.python.tool import PythonREPLTool
 #from langchain_community.tools.python_repl import PythonREPLTool
 from langchain_experimental.tools.python.tool import PythonREPLTool
@@ -117,7 +118,12 @@ image_tool = Tool(
 
 repl_tool = PythonREPLTool()
 hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
-llm = HuggingFaceHub(repo_id="google/flan-t5-xl", huggingfacehub_api_token=hf_token,model_kwargs={"temperature": 0.2, "max_length": 1024})
+#llm = HuggingFaceHub(repo_id="google/flan-t5-xl", huggingfacehub_api_token=hf_token,model_kwargs={"temperature": 0.2, "max_length": 1024})
+llm = HuggingFaceEndpoint(
+    repo_id="google/flan-t5-xl",
+    huggingfacehub_api_token=hf_token,
+    model_kwargs={"temperature": 0.2, "max_length": 1024}
+)
 
 tools = [wiki_tool, youtube_tool, audio_tool, excel_tool, image_tool, repl_tool]
 agent_instance = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)