Update app.py

app.py

@@ -233,13 +233,13 @@ def get_detection_data(image_data):
             nested_dict = image_data.get("image", {}).get("image", {})
             full_data_url = nested_dict.get("data", "")
             # get model size and confidence threshold
-            model_size = image_data.get("model_size", "Small - Better performance and less accuracy")
-            confidence_threshold = image_data.get("confidence_threshold", 0.6)  # Default to 60%
+            #model_size = image_data.get("model_size", "Small - Better performance and less accuracy")
+            #confidence_threshold = image_data.get("confidence_threshold", 0.6)  # Default to 60%
         else:
             full_data_url = image_data
             
-            model_size = "Small - Better performance and less accuracy"  # Fallback default
-            confidence_threshold = 0.6  # Fallback default
+            #model_size = "Small - Better performance and less accuracy"  # Fallback default
+            #confidence_threshold = 0.6  # Fallback default
 
         if not full_data_url:
             return {"error": "No base64 data found in input."}
@@ -254,17 +254,6 @@ def get_detection_data(image_data):
         img = np.array(img)
         img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
 
-        # Log received values for reference (settings applied by prior API calls)
-        print(f"Received settings from payload (reference) - Model Size: {model_size}, Confidence Threshold: {confidence_threshold}")
-        # Verify actual models and threshold used
-        expected_yolo = "yolov8s-seg" if model_size == "Small - Better performance and less accuracy" else \
-                       "yolov8m-seg" if model_size == "Medium - Balanced performance and accuracy" else "yolov8l-seg"
-        expected_midas = "midas_v21_small_256" if model_size == "Small - Better performance and less accuracy" else \
-                        "dpt_hybrid_384" if model_size == "Medium - Balanced performance and accuracy" else "dpt_large_384"
-        print(f"Expected YOLO model: {expected_yolo}, Actual YOLO model: {img_seg.model_type}")
-        print(f"Expected MiDaS model: {expected_midas}, Actual MiDaS model: {depth_estimator.model_type}")
-        print(f"Actual confidence threshold: {img_seg.confidence_threshold}")
-
         #image = utils.resize(img)
         resized_image = utils.resize(img) #depth requires resizing 
         print(f"Debug - Resized image shape: {resized_image.shape}")