From 5beb03474237a056e83895ca51800d0589577fea Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Fri, 15 Mar 2024 02:42:09 +0100 Subject: [PATCH] add warning --- .../java/ai/nets/samj/communication/model/EfficientSAM.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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