fix: import facemeshdetector and remove logger

app.py

@@ -11,9 +11,10 @@ import gc
 import math
 import cvzone
 from cvzone.PoseModule import PoseDetector
+from cvzone.FaceMeshModule import FaceMeshDetector
 import spaces
 
-choker_images = [Image.open(os.path.join("choker", x)) for x in os.listdir("choker")]
+choker_images = [Image.open(os.path.join("short_necklace", x)) for x in os.listdir("short_necklace")]
 person_images = [Image.open(os.path.join("without_necklace", x)) for x in os.listdir("without_necklace")]
 
 model_id = "stabilityai/stable-diffusion-2-inpainting"
@@ -38,8 +39,6 @@ def clothing_try_on_n_necklace_try_on(image, jewellery):
     copy_image = image.copy()
     jewellery = np.array(jewellery)
 
-    logger.info(f"NECKLACE TRY ON :: detecting pose and landmarks :: {storename}")
-
     image = detector.findPose(image)
     lmList, _ = detector.findPosition(image, bboxWithHands=False, draw=False)
 
@@ -51,7 +50,6 @@ def clothing_try_on_n_necklace_try_on(image, jewellery):
     landmarksDistance = int(
         ((leftLandmark[0] - rightLandmark[0]) ** 2 + (leftLandmark[1] - rightLandmark[1]) ** 2) ** 0.5)
 
-    logger.info(f"NECKLACE TRY ON :: estimating neck points")
 
     avg_x1 = int(leftLandmark[0] - landmarksDistance * 0.12)
     avg_x2 = int(rightLandmark[0] + landmarksDistance * 0.12)
@@ -59,7 +57,6 @@ def clothing_try_on_n_necklace_try_on(image, jewellery):
     avg_y1 = int(leftLandmark[1] + landmarksDistance * 0.5)
     avg_y2 = int(rightLandmark[1] + landmarksDistance * 0.5)
 
-    logger.info(f"NECKLACE TRY ON :: scaling the necklace image ")
 
     if avg_y2 < avg_y1:
         angle = math.ceil(
@@ -79,7 +76,6 @@ def clothing_try_on_n_necklace_try_on(image, jewellery):
     origImgRatio = xdist / jewellery.shape[1]
     ydist = jewellery.shape[0] * origImgRatio
 
-    logger.info(f"NECKLACE TRY ON :: adding offset based on the necklace shape")
 
     image_gray = cv2.cvtColor(jewellery, cv2.COLOR_BGRA2GRAY)
     for offset_orig in range(image_gray.shape[1]):
@@ -94,9 +90,6 @@ def clothing_try_on_n_necklace_try_on(image, jewellery):
     )
     jewellery = cvzone.rotateImage(jewellery, angle)
     y_coordinate = avg_y1 - offset
-
-    logger.info(f"NECKLACE TRY ON :: generating output :: {storename}")
-
     result = cvzone.overlayPNG(copy_image, jewellery, (avg_x1, y_coordinate))
 
     blackedNecklace = np.zeros(shape=copy_image.shape)
@@ -176,4 +169,3 @@ with gr.Blocks() as interface:
     submit.click(fn=clothing_try_on_n_necklace_try_on, inputs=[inputImage, selectedNecklace], outputs=[outputOne])
 
 interface.launch(debug=True)
-