check.py

from handler import EndpointHandler

my_handler = EndpointHandler()
payload = {"inputs": {"prompt": "a dog waiting for its companion to come."}}

# test the handler
image = my_handler(payload)
image.save("image.png")