Update controller.py

controller.py

@@ -9,6 +9,7 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
     log_response("url_endpoint \n {}".format(url_endpoint))
     log_response("url_endpoint \n {}".format(document))
     log_response("url_endpoint \n {}".format(image))
+    image = image
     
     agent = CustomHfAgent(
         url_endpoint=url_endpoint,
@@ -17,7 +18,7 @@ def handle_submission(user_message, selected_tools, url_endpoint, document, imag
         input_params={"max_new_tokens": 192},
     )
 
-    response = agent.chat(user_message)
+    response = agent.chat(user_message,document=document,image=image)
 
     log_response("Agent Response\n {}".format(response))