Update object_detection.py

object_detection.py

@@ -43,7 +43,7 @@ def detectObjects(imageFile, confidence_score):
     if not matches:
         raise FileNotFoundError(f"No output image found for {stem} in {pred_dir}")
     # Load and display the image
-    img = mpimg.imread(image_path)
+    img = mpimg.imread(str(matches[0]))
     return img
     #plt.figure(figsize=(8, 8))
     #plt.imshow(img)