test_handler.py

import base64
from io import BytesIO
import requests
from PIL import Image

# Import the EndpointHandler class
from handler import EndpointHandler

# Fetch the image from the URL
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
response = requests.get(url, stream=True)
image = Image.open(response.raw).convert("RGB")  # Ensure the image is in RGB format

# Encode the image as base64
buffered = BytesIO()
image.save(buffered, format="JPEG")  # Save the image to a buffer in JPEG format
encoded_image = base64.b64encode(buffered.getvalue()).decode("utf-8")

# Prepare the input data
data = {"images": [encoded_image]}

# Initialize the handler
handler = EndpointHandler()

# Process the image
result = handler(data)

# Print the result
print(result)