jasper-asr/jasper/data_utils/data_server.py

30 lines
586 B
Python

import typer
import rpyc
import os
from pathlib import Path
from rpyc.utils.server import ThreadedServer
app = typer.Typer()
class ASRDataService(rpyc.Service):
def get_data_loader(self, data_manifest: Path):
return "hello"
@app.command()
def run_server(port: int = 0):
listen_port = port if port else int(os.environ.get("ASR_RPYC_PORT", "8044"))
service = ASRDataService()
t = ThreadedServer(service, port=listen_port)
typer.echo(f"starting asr server on {listen_port}...")
t.start()
def main():
app()
if __name__ == "__main__":
main()