diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index ac2081fad..434e8b500 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -295,8 +295,14 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi): EventMetadataTypeEnum.sub_label, (obj_data["id"], sub_label, score), ) - elif self.model_config.object_config.classification_type == ObjectClassificationType.attribute: - pass + elif ( + self.model_config.object_config.classification_type + == ObjectClassificationType.attribute + ): + self.sub_label_publisher.publish( + EventMetadataTypeEnum.attribute, + (obj_data["id"], self.model_config.name, sub_label, score), + ) def handle_request(self, topic, request_data): if topic == EmbeddingsRequestEnum.reload_classification_model.value: