From ab3567aac63b3b26606c4d5b670b71723682bb5a Mon Sep 17 00:00:00 2001 From: Bingsu Date: Tue, 30 May 2023 09:39:28 +0900 Subject: [PATCH] fix: mediapipe partial error --- adetailer/mediapipe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/adetailer/mediapipe.py b/adetailer/mediapipe.py index dfaca74..f5448f2 100644 --- a/adetailer/mediapipe.py +++ b/adetailer/mediapipe.py @@ -13,8 +13,8 @@ def mediapipe_predict( model_type: str, image: Image.Image, confidence: float = 0.3 ) -> PredictOutput: mapping = { - "mediapipe_face_short": partial(mediapipe_face_detection, model_type=0), - "mediapipe_face_full": partial(mediapipe_face_detection, model_type=1), + "mediapipe_face_short": partial(mediapipe_face_detection, 0), + "mediapipe_face_full": partial(mediapipe_face_detection, 1), "mediapipe_face_mesh": mediapipe_face_mesh, } if model_type in mapping: