Fixed returns

This commit is contained in:
DESKTOP-UL3K6JV\Daniq
2024-06-16 14:55:56 +05:00
parent 666245a486
commit c83cb1ba5b
+2 -2
View File
@@ -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,)