predict.py

#!python

import sys
import importlib
from pathlib import Path

models = set([
    x.name for x in Path('asr/models').iterdir()
    if x.is_dir() and x.name[0] not in ['_', '.']
])

try:
    model, argv = sys.argv[1], sys.argv[2:]
    if model not in models:
        raise
except:
    print(f"Error: choose one of models in {models}")
    sys.exit(1)

try:
    m = importlib.import_module(f"asr.models.{model}")
    m.predict(argv)
except:
    raise