Load the model with [AutoModelForObjectDetection] using the same checkpoint as in the preprocessing.