Named volume for db, --update-models arg

This commit is contained in:
Piero Toffanin 2022-12-20 11:13:56 -05:00
commit 959f4638cc
6 changed files with 20 additions and 18 deletions

View file

@ -100,7 +100,7 @@ def get_routes_limits(default_req_limit, daily_req_limit, api_keys_db):
def create_app(args):
from app.init import boot
boot(args.load_only)
boot(args.load_only, args.update_models)
from app.language import load_languages

View file

@ -156,6 +156,11 @@ _default_options_objects = [
'default_value': False,
'value_type': 'bool'
},
{
'name': 'UPDATE_MODELS',
'default_value': False,
'value_type': 'bool'
},
]

View file

@ -5,9 +5,9 @@ from argostranslate import package, translate
import app.language
def boot(load_only=None):
def boot(load_only=None, update_models=False):
try:
check_and_install_models(load_only_lang_codes=load_only)
check_and_install_models(force=update_models, load_only_lang_codes=load_only)
except Exception as e:
print("Cannot update models (normal if you're offline): %s" % str(e))

View file

@ -144,7 +144,9 @@ def get_args():
parser.add_argument(
"--disable-web-ui", default=DEFARGS['DISABLE_WEB_UI'], action="store_true", help="Disable web ui"
)
parser.add_argument(
"--update-models", default=DEFARGS['UPDATE_MODELS'], action="store_true", help="Update language models at startup"
)
return parser.parse_args()