fixing bug #871

MouseLand · Feb 22, 2024 · b138040 · b138040
1 parent f744fe0
commit b138040
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/cellpose/gui/gui.py b/cellpose/gui/gui.py
@@ -981,7 +981,7 @@ def model_choose(self, custom=False):
             if custom:
                 model_name = self.ModelChooseC.currentText()
             else:
-                model_name = self.ModelChooseB.currentText()
+                model_name = self.net_names[index - 1]
             print(f"GUI_INFO: selected model {model_name}, loading now")
             self.initialize_model(model_name=model_name, custom=custom)
             self.diameter = self.model.diam_labels

diff --git a/cellpose/models.py b/cellpose/models.py
@@ -41,7 +41,6 @@
     "invert": False
 }
 
-
 def model_path(model_type, model_index=0):
     torch_str = "torch"
     if model_type == "cyto" or model_type == "cyto2" or model_type == "nuclei":