Update app.py

app.py

@@ -6,8 +6,9 @@ from PIL import Image
 import cv2
 import mediapipe as mp
 
+# Initialize Mediapipe Pose
 mp_pose = mp.solutions.pose
-mp_drawing = mp.solutions.drawing_utils
+pose = mp_pose.Pose(static_image_mode=True, model_complexity=2)
 
 # Load sample product data
 product_data = pd.DataFrame({
@@ -18,14 +19,13 @@ product_data = pd.DataFrame({
 })
 
 def get_keypoints(image):
-    with mp_pose.Pose(static_image_mode=True, model_complexity=2) as pose:
-        results = pose.process(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-        if not results.pose_landmarks:
-            return None
-        keypoints = {}
-        for idx, lm in enumerate(results.pose_landmarks.landmark):
-            keypoints[idx] = (int(lm.x * image.shape[1]), int(lm.y * image.shape[0]))
-        return keypoints
+    results = pose.process(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    if not results.pose_landmarks:
+        return None
+    keypoints = {}
+    for idx, lm in enumerate(results.pose_landmarks.landmark):
+        keypoints[idx] = (int(lm.x * image.shape[1]), int(lm.y * image.shape[0]))
+    return keypoints
 
 def process_image(image, product):
     # Convert the uploaded image to a PIL image