Update app.py

app.py

@@ -7,6 +7,9 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+from huggingface_hub import login
+import os
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -47,6 +50,8 @@ custom_role_conversions=None,
 )
 
 
+login(token=os.getenv("HF_TOKEN"))
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)