30 lines
586 B
Python
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()
|