Added GLiNER to download_models

Signed-off-by: Marcos Martínez Galindo <marcosmartinezgalindo@Marcoss-MacBook-Pro.local>
IBM · Aug 15, 2024 · 49e4b96 · 49e4b96
1 parent 6af861e
commit 49e4b96
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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: