Update object_detection.py

object_detection.py

@@ -10,13 +10,13 @@ import time
 # Method to detect objects with given confidence score
 
 def detectObjects(imageFile, confidence_score):
-    #current_path = os.getcwd()
-    print(f"Current working directory: {imageFile}")
+    current_path = os.getcwd()
+    print(f"Current working directory: {current_path}")
     # Load a pre-trained YOLOv8 model (e.g., YOLOv8n)
     model = YOLO("yolov8n.pt", verbose=False)  # Use other variants like yolov8s.pt for better accuracy
     
     # Define a custom directory to save the results
-    custom_save_dir = f"./"
+    custom_save_dir = f"{current_path}/runs/detect/predict"
 
     custom_read_dir = "./"