fix: fix object classification model not reload

This commit is contained in:
GuoQing Liu 2026-02-12 23:47:03 +08:00 committed by GitHub
parent 6accc38275
commit ceb9e8c044
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -658,6 +658,7 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
def handle_request(self, topic, request_data): def handle_request(self, topic, request_data):
if topic == EmbeddingsRequestEnum.reload_classification_model.value: if topic == EmbeddingsRequestEnum.reload_classification_model.value:
if request_data.get("model_name") == self.model_config.name: if request_data.get("model_name") == self.model_config.name:
self.__build_detector()
logger.info( logger.info(
f"Successfully loaded updated model for {self.model_config.name}" f"Successfully loaded updated model for {self.model_config.name}"
) )