Added try/except on load models

Signed-off-by: Marcos Martinez <[email protected]>
IBM · Sep 4, 2023 · 5ab08ac · 5ab08ac
1 parent 7c1bd58
commit 5ab08ac
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/zshot/utils/download_models.py b/zshot/utils/download_models.py
@@ -6,9 +6,18 @@
 
 
 def load_all():
-    load_few_rel_zs()
-    load_medmentions_zs()
-    load_ontonotes_zs()
+    try:
+        load_few_rel_zs()
+    except ConnectionError:
+        pass
+    try:
+        load_medmentions_zs()
+    except ConnectionError:
+        pass
+    try:
+        load_ontonotes_zs()
+    except ConnectionError:
+        pass
     try:
         LinkerSMXM().load_models()
     except RuntimeError: