♻️ Use models from IBM org on HG hub

Signed-off-by: Gabriele Picco <piccogabriele@gmail.com>
IBM · Oct 12, 2022 · bed746e · bed746e
1 parent 237d5d9
commit bed746e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/zshot/linker/linker_regen/linker_regen.py b/zshot/linker/linker_regen/linker_regen.py
@@ -8,7 +8,7 @@
 from zshot.linker.linker_regen.utils import create_input
 from zshot.utils.data_models import Entity, Span
 
-MODEL_NAME = "gabriele-picco/regen-disambiguation"
+MODEL_NAME = "ibm/regen-disambiguation"
 
 START_ENT_TOKEN = "[START_ENT]"
 END_ENT_TOKEN = "[END_ENT]"

diff --git a/zshot/linker/linker_smxm.py b/zshot/linker/linker_smxm.py
@@ -19,7 +19,7 @@
 )
 from zshot.utils.data_models import Span
 
-MODEL_NAME = "gabriele-picco/smxm"
+MODEL_NAME = "ibm/smxm"
 
 
 class LinkerSMXM(Linker):