diff --git a/python/dolma/taggers/code_composition.py b/python/dolma/taggers/code_composition.py index c8198b41..f1782934 100644 --- a/python/dolma/taggers/code_composition.py +++ b/python/dolma/taggers/code_composition.py @@ -24,7 +24,7 @@ @TaggerRegistry.add("code_composition") class CodeProseCompositionClassifier(BaseFastTextTagger): - MODEL_PATH = "hf://techarb/code-prose-composition/code-comment-prose-model.bin" # noqa: E501 + MODEL_PATH = "https://huggingface.co/techarb/code-prose-composition/resolve/main/code-comment-prose-model.bin" # noqa: E501 def __init__(self): super().__init__(model_path=self.MODEL_PATH, model_mode=self.DOCUMENT_LEVEL_TAGGER)