diff --git a/classifiers.py b/classifiers.py index 82f7557..cf20c51 100644 --- a/classifiers.py +++ b/classifiers.py @@ -39,7 +39,7 @@ class HFClassify: } } - RETURN_TYPES = ("FLOAT", "INT"") + RETURN_TYPES = ("FLOAT", "INT") RETURN_NAMES = ("raw_score", "rounded_score") FUNCTION = "classify"