from serve_roformer import RoformerService

def main():
    print("Hello from model-yolo-person-classify-market!")

    # uv run python person_attr.py --image_url "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg" --xywh "404,290,74,193"

    roformer_service = RoformerService()
    roformer_service.load("net_last.pth")
    result = roformer_service.predict({
        "image_url": "https://acai.ketidev.kr:20443/detect/image/202606/20260619_145116_image.jpg",
        "xywh": "404,290,74,193"
    })
    print(result)


if __name__ == "__main__":
    main()