Update model.py

model.py

@@ -23,7 +23,7 @@ class Model:
 
 
     def _load_model(self, name: str) -> nn.Module:
-        return init_detector('configs/_base_/faster-rcnn_r50_fpn_1x_coco.py', 'models/orgaquant_pretrained.pth' , device=self.device)
+        return init_detector('configs/faster-rcnn_r50_organoid_orgaquant.py', 'models/orgaquant_pretrained.pth' , device=self.device)
 
     def set_model(self, name: str) -> None:
         if name == self.model_name:
@@ -36,9 +36,7 @@ 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
 
@@ -46,9 +44,7 @@ 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(