diff --git a/models/translation/translation_translator.py b/models/translation/translation_translator.py index 21710e2d..8c3e401b 100644 --- a/models/translation/translation_translator.py +++ b/models/translation/translation_translator.py @@ -6,11 +6,6 @@ from .translation_languages import translation_lang import ctranslate2 import transformers -TRANSLATE_MODELS = { - "small": "facebook/m2m100_418M", - "large": "facebook/m2m100_1.2B" -} - # Translator class Translator(): def __init__(self, path, weight_config): diff --git a/models/translation/utils.py b/models/translation/utils.py index c41fbb88..f1c40b9e 100644 --- a/models/translation/utils.py +++ b/models/translation/utils.py @@ -2,7 +2,7 @@ import tempfile from zipfile import ZipFile from os import path as os_path from os import makedirs as os_makedirs -from requests import get as requests_get, head as requests_head +from requests import get as requests_get from tqdm import tqdm from typing import Callable import hashlib