You can add the function that verify is data is None like:

Create a PIL image from the NumPy array

image = Image.fromarray(arry)

if image is None:
    pass