from transformers import pipeline

def model_fn(model_dir):
    pipe = pipeline("object-detection", model=model_dir, threshold=0.1)
    return pipe