Update custom_agent.py

custom_agent.py

@@ -5,7 +5,7 @@ import io
 import requests
 import time
 from transformers import Agent
-
+from logger import log_response
 
 import time  
 import torch