From 49e4b96ff9b40520a19d54050fa8383e09fa5f9a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marcos=20Mart=C3=ADnez=20Galindo?= Date: Thu, 15 Aug 2024 10:41:21 +0100 Subject: [PATCH] Added GLiNER to download_models MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Marcos Martínez Galindo --- zshot/utils/download_models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/zshot/utils/download_models.py b/zshot/utils/download_models.py index 15ae06b..283140e 100644 --- a/zshot/utils/download_models.py +++ b/zshot/utils/download_models.py @@ -1,5 +1,5 @@ from zshot.evaluation.dataset import load_few_rel_zs, load_medmentions_zs, load_ontonotes_zs -from zshot.linker import LinkerRegen, LinkerSMXM, LinkerTARS +from zshot.linker import LinkerRegen, LinkerSMXM, LinkerTARS, LinkerGLINER from zshot.mentions_extractor import MentionsExtractorFlair from zshot.mentions_extractor.utils import ExtractorType from zshot.relation_extractor.relation_extractor_zsrc import RelationsExtractorZSRC @@ -30,6 +30,10 @@ def load_all(): LinkerTARS().load_models() except RuntimeError: pass + try: + LinkerGLINER().load_models() + except RuntimeError: + pass try: RelationsExtractorZSRC().load_models() except RuntimeError: