change name of function to match main- also edit so channels are extr…

…acted
AllenCell · Feb 12, 2024 · 84ca73c · 84ca73c
1 parent c1d1314
commit 84ca73c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/allencell_ml_segmenter/prediction/file_input_widget.py b/src/allencell_ml_segmenter/prediction/file_input_widget.py
@@ -121,7 +121,7 @@ def __init__(self, model: PredictionModel):
 
         self._browse_dir_edit: InputButton = InputButton(
             self._model,
-            lambda dir: self._model.set_input_image_path(Path(dir)),
+            lambda dir: self._model.set_input_image_path(Path(dir), extract_channels=True),
             "Select directory...",
             FileInputMode.DIRECTORY_OR_CSV,
         )