diff --git a/src/main/java/ai/nets/samj/gui/ModelSelection.java b/src/main/java/ai/nets/samj/gui/ModelSelection.java index e7bc442..5614d29 100644 --- a/src/main/java/ai/nets/samj/gui/ModelSelection.java +++ b/src/main/java/ai/nets/samj/gui/ModelSelection.java @@ -44,7 +44,14 @@ protected static ModelSelection create(List models, ModelSelectionList protected SAMModel getSelectedModel() { return selected; } - + + protected boolean isModelInstalled(String modelName) { + for (SAMModel m : this.models) { + if (m.getName().equals(modelName)) return m.isInstalled(); + } + return false; + } + protected JButton getButton() { return this.btn; }