diff --git a/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_face_detection.py b/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_face_detection.py index 99892e1..f9c47da 100755 --- a/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_face_detection.py +++ b/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_face_detection.py @@ -32,7 +32,11 @@ def face_detection(image): "image": img_base64 } - response = requests.post(api_url, json=payload) + headers = { + "Authorization": f"Bearer {api_access_key}" + } + + response = requests.post(api_url, json=payload, headers=headers) if response.status_code == 200: result = response.json() @@ -55,7 +59,7 @@ def image_callback(msg): def main(): - global api_url, pub + global api_url, api_access_key, pub config_file_path = os.path.join(os.path.dirname(__file__), '../config/configs.json') @@ -67,6 +71,7 @@ def main(): config = json.load(config_file) url = config['url'] endpoint_path = config['endpoint_path']["face_detection"] + api_access_key = config["Bellande_Framework_Access_Key"] # Initialize ROS node if ros_version == "1": diff --git a/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_prediction.py b/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_prediction.py index 1af5382..77f5580 100755 --- a/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_prediction.py +++ b/ros_web_api_bellande_2d_computer_vision/src/bellande_2d_computer_vision_prediction.py @@ -34,7 +34,11 @@ def predict(image): "image": img_base64 } - response = requests.post(api_url, json=payload) + headers = { + "Authorization": f"Bearer {api_access_key}" + } + + response = requests.post(api_url, json=payload, headers=headers) if response.status_code == 200: result = response.json() @@ -67,6 +71,7 @@ def main(): config = json.load(config_file) url = config['url'] endpoint_path = config['endpoint_path']["prediction"] + api_access_key = config["Bellande_Framework_Access_Key"] # Initialize ROS node if ros_version == "1":