Update agent.py

agent.py

@@ -54,6 +54,9 @@ from langchain.chains import LLMChain
 from langchain.agents import initialize_agent, Tool, AgentType
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import login
+
 
 load_dotenv()