diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java b/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java index a693872..6d1cd83 100644 --- a/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java +++ b/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java @@ -64,6 +64,8 @@ public class EfficientSAM implements SAMModel { + "Paper: EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment\n" + "Anything"; + private static final String CAUTION_STRING = "

CAUTION: This model is computationally heavy. It is not recommended to use it on lower-end computers.

"; + public EfficientSAM() {} /** @@ -102,7 +104,7 @@ public String getName() { * {@inheritDoc} */ public String getDescription() { - return HTML_DESCRIPTION + (!this.installed ? SAMModel.HTML_NOT_INSTALLED : ""); + return HTML_DESCRIPTION + (!this.installed ? SAMModel.HTML_NOT_INSTALLED : "") + CAUTION_STRING; } @Override