Upload app.py

app.py

@@ -1,4 +1,4 @@
-import os
+import os, subprocess
 import spaces
 import gradio as gr
 from functools import partial
@@ -20,6 +20,8 @@ from PIL import Image, ImageDraw
 from torchvision.transforms.functional import to_pil_image
 from ultralytics import YOLO
 
+subprocess.run("pip list", shell=True)
+
 def create_mask_from_bbox(
     bboxes: np.ndarray, shape: tuple[int, int]
 ) -> list[Image.Image]:
@@ -90,8 +92,14 @@ def infer(text: str):
     person_model_path = hf_hub_download("Bingsu/adetailer", "person_yolov8s-seg.pt")
     person_detector = partial(yolo_detector, model_path=person_model_path)
     detectors.append(person_detector)
-
-    init_image = Image.open("./image.webp")
+    face_model_path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
+    face_detector = partial(yolo_detector, model_path=face_model_path)
+    detectors.append(face_detector)
+    hand_model_path = hf_hub_download("Bingsu/adetailer", "hand_yolov8n.pt")
+    hand_detector = partial(yolo_detector, model_path=hand_model_path)
+    detectors.append(hand_detector)
+    init_image = Image.open("./image.webp").convert("RGB")
+    masks = None
     for j, detector in enumerate(detectors):
         masks = detector(init_image)
     return str(masks)