Update model.py

model.py

@@ -36,7 +36,9 @@ class Model:
     ) -> tuple[list[np.ndarray] | tuple[list[np.ndarray],
                                         list[list[np.ndarray]]]
                | dict[str, np.ndarray], np.ndarray]:
+        print('Image1', image.shape)
         out = self.detect(image)
+        print('Out1', out)
         vis = self.visualize_detection_results(image, out, score_threshold)
         return out, vis
 
@@ -44,7 +46,9 @@ class Model:
         self, image: np.ndarray
     ) -> list[np.ndarray] | tuple[
             list[np.ndarray], list[list[np.ndarray]]] | dict[str, np.ndarray]:
+        print('Image2', image.shape)
         out = inference_detector(self.model, image)
+        print('Out2',out)
         return out
 
     def visualize_detection_results(