From ae4ffeb5ab6a1d0dc62ff4852b4c7bb37839d1d6 Mon Sep 17 00:00:00 2001 From: Ian Coccimiglio Date: Sun, 28 Apr 2024 20:29:35 -0700 Subject: [PATCH] Fixed tab/spacing --- .../minimal_detection/prompt_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/napari_segment_everything/minimal_detection/prompt_generator.py b/src/napari_segment_everything/minimal_detection/prompt_generator.py index 02898bd..e89da6b 100644 --- a/src/napari_segment_everything/minimal_detection/prompt_generator.py +++ b/src/napari_segment_everything/minimal_detection/prompt_generator.py @@ -109,8 +109,8 @@ class RcnnDetector(BaseDetector): def __init__(self, model_path, device, trainable=True): super().__init__(model_path, trainable) self.model_type = "FasterRCNN" - if device == "mps": - device = "cpu" + if device == "mps": + device = "cpu" self.device = device self.model = fasterrcnn_mobilenet_v3_large_fpn( box_detections_per_img=500,