Update pipeline/logic.py

pipeline/logic.py

@@ -24,6 +24,8 @@ def download_image_from_url(image_url: str) -> Image.Image:
     print(" [Download] ✅ Image downloaded and standardized to RGB.")
     return image_rgb
 
+# In pipeline/logic.py
+
 def detect_and_crop(image: Image.Image, object_name: str, models: dict) -> Image.Image:
     print(f"\n [Detect & Crop] Starting detection for object: '{object_name}'")
     image_np = np.array(image.convert("RGB"))
@@ -32,9 +34,16 @@ def detect_and_crop(image: Image.Image, object_name: str, models: dict) -> Image
     inputs = models['processor_gnd'](images=image, text=prompt, return_tensors="pt").to(models['device'])
     with torch.no_grad():
         outputs = models['model_gnd'](**inputs)
+
+    # THIS IS THE CORRECTED LINE:
     results = models['processor_gnd'].post_process_grounded_object_detection(
-        outputs, inputs.input_ids, box_threshold=0.4, text_threshold=0.3, target_sizes=[(height, width)]
+        outputs,
+        inputs.input_ids,
+        object_detection_threshold=0.4, # Renamed from box_threshold
+        text_detection_threshold=0.3,   # Renamed from text_threshold
+        target_sizes=[(height, width)]
     )
+    
     if not results or len(results[0]['boxes']) == 0:
         print(" [Detect & Crop] ⚠ Warning: Grounding DINO did not detect the object. Using full image.")
         return image