Update main.py

main.py

@@ -5,7 +5,7 @@ os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf"
 os.environ["HF_HOME"] = "/tmp/hf"
 
 from langchain.llms import HuggingFacePipeline
-from langchain.agents import initialize_agent, AgentType
+from langchain.agents.agent_types import AgentType
 from langchain.tools import BaseTool
 from transformers import pipeline