diff --git a/core/src/main/java/com/dite/znpt/config/YoloModelRegistry.java b/core/src/main/java/com/dite/znpt/config/YoloModelRegistry.java index d820dd0..8d3584c 100644 --- a/core/src/main/java/com/dite/znpt/config/YoloModelRegistry.java +++ b/core/src/main/java/com/dite/znpt/config/YoloModelRegistry.java @@ -43,10 +43,10 @@ public class YoloModelRegistry { @PostConstruct public void loadModelsOnStartup() throws OrtException { -// List configs = modelConfigMapper.selectList(Wrappers.emptyWrapper()); -// for (ModelConfigEntity config : configs) { -// loadModel(config); -// } + List configs = modelConfigMapper.selectList(Wrappers.emptyWrapper()); + for (ModelConfigEntity config : configs) { + loadModel(config); + } } public void loadModel(ModelConfigEntity config) throws OrtException {