File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -872,12 +872,13 @@ def run(self, workspace):
872872
873873 elif self .cellpose_version .value == 'v4' :
874874 assert int (self .cellpose_ver [0 ])== 4 , "Cellpose version selected in RunCellpose module doesn't match version in Python"
875+ LOGGER .info (f"Loading new model: { self .mode .value } " )
875876 if self .mode .value == 'custom' :
876877 model_file , model_directory , model_path = get_custom_model_vars (self )
877- model_params = ( self .mode . value , self . use_gpu . value )
878- LOGGER . info ( f"Loading new model: { self .mode .value } " )
879- self . current_model = models . CellposeModel ( gpu = self . use_gpu . value )
880- self .current_model_params = model_params
878+ self .current_model = models . CellposeModel (
879+ pretrained_model = model_path , gpu = self .use_gpu .value )
880+ else :
881+ self .current_model = models . CellposeModel ( gpu = self . use_gpu . value )
881882
882883 if self .use_gpu .value :
883884 try :
You can’t perform that action at this time.
0 commit comments