Update animal model name

modules/live_portrait/model_downloader.py

@@ -20,11 +20,11 @@ MODELS_ANIMAL_URL = {
     "warping_module": "https://huggingface.co/Kijai/LivePortrait_safetensors/resolve/main/animal/warping_module.safetensors",
     "spade_generator": "https://huggingface.co/Kijai/LivePortrait_safetensors/resolve/main/animal/spade_generator.safetensors",
     "stitching_retargeting_module": "https://huggingface.co/Kijai/LivePortrait_safetensors/resolve/main/animal/stitching_retargeting_module.safetensors",
-    "face_yolov8n": "https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8n.pt"
+    # Just animal detection model not the face, needs better model
+    "yolo_v5s_animal_det": "https://huggingface.co/degirum/yolo_v5s_animal_det_512x512_quant_n2x_cpu_1/resolve/main/yolo_v5s_animal_det_512x512_quant_n2x_cpu_1.n2x"
 }
 
 
-
 def download_model(
     file_path: str,
     url: str,

modules/utils/paths.py

@@ -23,7 +23,8 @@ MODEL_ANIMAL_PATHS = {
     "warping_module": os.path.join(MODELS_ANIMAL_DIR, "warping_module.safetensors"),
     "spade_generator": os.path.join(MODELS_ANIMAL_DIR, "spade_generator.safetensors"),
     "stitching_retargeting_module": os.path.join(MODELS_ANIMAL_DIR, "stitching_retargeting_module.safetensors"),
-    "face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt")
+    # Just animal detection model not the face, needs better model
+    "yolo_v5s_animal_det": os.path.join(MODELS_ANIMAL_DIR, "yolo_v5s_animal_det.n2x")
 }
 MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png")
 I18N_YAML_PATH = os.path.join(PROJECT_ROOT_DIR, "i18n", "translation.yaml")