Update custom_agent.py

custom_agent.py

@@ -6,6 +6,10 @@ import requests
 import time
 from transformers import Agent
 
+
+import time  
+import torch
+
 class CustomHfAgent(Agent):
     def __init__(self, url_endpoint, token, chat_prompt_template=None, run_prompt_template=None, additional_tools=None, input_params=None):
         super().__init__(