Update app.py

app.py

@@ -10,16 +10,26 @@ from flask_cors import CORS
 import os
 import shutil
 
-# NEW: Set mediapipe model path to avoid permission issues
-# NEW: Define custom model path
+# --- Monkey Patch: Override model path before using mediapipe.pose ---
+from mediapipe.python.solutions import pose as mp_pose_module
+
 CUSTOM_MODEL_PATH = "/app/models/pose_landmark_heavy.tflite"
-DEFAULT_MODEL_PATH = "/usr/local/lib/python3.10/site-packages/mediapipe/modules/pose_landmark/pose_landmark_heavy.tflite"
 
-# NEW: Replace the model file if not already done
-if not os.path.exists(DEFAULT_MODEL_PATH):
-    os.makedirs(os.path.dirname(DEFAULT_MODEL_PATH), exist_ok=True)
-    shutil.copy(CUSTOM_MODEL_PATH, DEFAULT_MODEL_PATH)
-    
+def monkey_patch_pose_path(model_complexity):
+    model_complexity_map = {
+        0: "pose_landmark_lite",
+        1: "pose_landmark_full",
+        2: "pose_landmark_heavy"
+    }
+    if model_complexity in model_complexity_map:
+        file_name = model_complexity_map[model_complexity] + ".tflite"
+        mp_pose_module.MODEL_PATHS[model_complexity] = CUSTOM_MODEL_PATH
+    else:
+        raise ValueError(f"Unsupported model complexity: {model_complexity}")
+
+# Monkey patch before creating Pose instance
+monkey_patch_pose_path(model_complexity=2)
+  
 app = Flask(__name__)
 CORS(app)