From c83cb1ba5b6d6b8d5b30f232afc5b1ca24cc0027 Mon Sep 17 00:00:00 2001 From: "DESKTOP-UL3K6JV\\Daniq" <31.10.93@bk.ru> Date: Sun, 16 Jun 2024 14:55:56 +0500 Subject: [PATCH] Fixed returns --- classifiers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/classifiers.py b/classifiers.py index 8b61ec8..58d58e1 100644 --- a/classifiers.py +++ b/classifiers.py @@ -35,7 +35,7 @@ class YOLOClassifierModelLoader: model_path = folder_paths.get_full_path("ultralytics_classifiers", model_name) model = YOLO(model_path) - return model + return (model,) class YOLOClassify: @@ -62,4 +62,4 @@ class YOLOClassify: class_name = r.names[top1] break - return class_name + return (class_name,)