Fixed inputs

2024-06-16 14:53:22 +05:00
parent dc6cd3b3d1
commit 666245a486
1 changed files with 2 additions and 2 deletions
@@ -43,8 +43,8 @@ class YOLOClassify:
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "yolo_classifier_model": "YOLO_CLASSIFIER_MODEL",
-                "image_to_classify": "IMAGE"
+                "yolo_classifier_model": ("YOLO_CLASSIFIER_MODEL",),
+                "image_to_classify": ("IMAGE",)
            }
        }