diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index 5e35c6899..229383d9f 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -658,6 +658,7 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi): def handle_request(self, topic, request_data): if topic == EmbeddingsRequestEnum.reload_classification_model.value: if request_data.get("model_name") == self.model_config.name: + self.__build_detector() logger.info( f"Successfully loaded updated model for {self.model_config.name}" )