boolean updated

This commit is contained in:
David Strand
2024-12-18 19:33:31 +08:00
committed by GitHub
parent 4827437d37
commit dd0a0b63c6
+4 -4
View File
@@ -40,7 +40,7 @@ class HFClassify:
}
RETURN_TYPES = ("FLOAT", "INT", "BOOL") # Added BOOL for the new boolean output
RETURN_NAMES = ("raw_score", "rounded_score", "is_above_threshold") # Added name for the boolean output
RETURN_NAMES = ("raw_score", "rounded_score", "boolean") # Added name for the boolean output
FUNCTION = "classify"
def classify(self, hf_classifier_model, image_to_classify, target_label):
@@ -61,8 +61,8 @@ class HFClassify:
# Round the score to 0 or 1
rounded_score = int(round(score))
# Determine if the score is above the threshold of 0.5
is_above_threshold = score > 0.5
# Convert to boolean
boolean = bool(rounded_score)
# Return the raw score, rounded score, and the boolean indicating if above threshold
return (score, rounded_score, is_above_threshold)
return (score, rounded_score, boolean)