diff --git a/classifiers.py b/classifiers.py index e8c519d..91a504e 100644 --- a/classifiers.py +++ b/classifiers.py @@ -22,7 +22,7 @@ class YOLOClassifierModelLoader: def INPUT_TYPES(cls): return { "required": { - "model_name": cls.model_names + "model_name": (cls.model_names,) } }