Update app.py

app.py

@@ -6,41 +6,13 @@ import mediapipe as mp
 import torch
 from flask import Flask, request, jsonify
 import torch.nn.functional as F
-from flask_cors import CORS
-import os
-import shutil
 
-# --- 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"
-
-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)
 
 mp_pose = mp.solutions.pose
 mp_holistic = mp.solutions.holistic
-pose = mp_pose.Pose(
-    model_complexity=2,
-    static_image_mode=True,
-    min_detection_confidence=0.5,
-    min_tracking_confidence=0.5
-) # Improved accuracy
+pose = mp_pose.Pose(model_complexity=2)  # Improved accuracy
 holistic = mp_holistic.Holistic()  # For refining pose
 
 KNOWN_OBJECT_WIDTH_CM = 21.0  # A4 paper width in cm
@@ -399,4 +371,4 @@ def upload_images():
     })
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+    app.run(host='0.0.0.0', port=7860)