plume-asr/src/plume/cli/serve.py

17 lines
456 B
Python

import typer
from ..models.wav2vec2.serve import app as wav2vec2_app
from ..models.wav2vec2_transformers.serve import app as wav2vec2_transformers_app
from ..models.jasper_nemo.serve import app as jasper_app
app = typer.Typer()
app.add_typer(wav2vec2_app, name="wav2vec2")
app.add_typer(wav2vec2_transformers_app, name="wav2vec2_transformers")
app.add_typer(jasper_app, name="jasper")
@app.callback()
def serve():
"""
serve sub commands
"""