From bed746ec0f4df45d8d533869bffb93b94c52307c Mon Sep 17 00:00:00 2001 From: Gabriele Picco Date: Wed, 12 Oct 2022 16:16:27 +0100 Subject: [PATCH] :recycle: Use models from IBM org on HG hub Signed-off-by: Gabriele Picco --- zshot/linker/linker_regen/linker_regen.py | 2 +- zshot/linker/linker_smxm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/zshot/linker/linker_regen/linker_regen.py b/zshot/linker/linker_regen/linker_regen.py index 5e1ecb5..474bc58 100644 --- 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 index c2bc5c0..2a5ec64 100644 --- 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):