diff --git a/SuperTagger/SuperTagger.py b/SuperTagger/SuperTagger.py index 1108f2192fb47b10910ab840dfba79eaae82ad7c..31da8563915da53e19227e65fd377f34bea2a0ba 100644 --- a/SuperTagger/SuperTagger.py +++ b/SuperTagger/SuperTagger.py @@ -13,9 +13,9 @@ from tqdm import tqdm from transformers import AutoTokenizer from transformers import logging -from SuperTagger.Utils.SentencesTokenizer import SentencesTokenizer -from SuperTagger.Utils.SymbolTokenizer import SymbolTokenizer -from SuperTagger.Utils.Tagging_bert_model import Tagging_bert_model +from .Utils.SentencesTokenizer import SentencesTokenizer +from .Utils.SymbolTokenizer import SymbolTokenizer +from .Utils.Tagging_bert_model import Tagging_bert_model logging.set_verbosity(logging.ERROR) diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..d0947b7c6e7626450af84978c73c7d2d381bff19 --- /dev/null +++ b/__init__.py @@ -0,0 +1,2 @@ +from .SuperTagger.Utils import * +from .SuperTagger.SuperTagger import SuperTagger \ No newline at end of file