Skip to content

Commit 088d8ff

Browse files
authored
add import to solve check custom model (#262)
1 parent ff071a4 commit 088d8ff

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

active_plugins/runcellpose.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -441,6 +441,7 @@ def visible_settings(self):
441441

442442
def validate_module(self, pipeline):
443443
"""If using custom model, validate the model file opens and works"""
444+
from cellpose import models
444445
if self.mode.value == "custom":
445446
model_file = self.model_file_name.value
446447
model_directory = self.model_directory.get_absolute_path()

0 commit comments

Comments
 (0)