diff --git a/app/asr.py b/app/asr.py index f9ed401..d19c933 100644 --- a/app/asr.py +++ b/app/asr.py @@ -38,9 +38,9 @@ class NVidiaASR: } def init(): - asr_engine = NVidiaASR() g = ServerEnv() - g.generate = asr_engine.generate + g.asr_engine = NVidiaASR() + g.generate = g.asr_engine.generate if __name__ == '__main__': webapp(init) diff --git a/nvidia-asr.service b/nvidia-asr.service new file mode 100644 index 0000000..a566b8a --- /dev/null +++ b/nvidia-asr.service @@ -0,0 +1,14 @@ +[Unit] +Wants=systemd-networkd.service + +[Service] +User=ymq +Group=ymq +WorkingDirectory=/share/ymq/run/nvidia-asr +ExecStart=/share/ymq/run/nvidia-asr/start.sh +ExecStop=/share/ymq/run/nvidia-asr/stop.sh +StandardOutput=append:/var/log/asr/asr.log +StandardError=append:/var/log/asr/asr.log +SyslogIdentifier=asr +[Install] +WantedBy=multi-user.target diff --git a/start.sh b/start.sh new file mode 100644 index 0000000..18d6451 --- /dev/null +++ b/start.sh @@ -0,0 +1,5 @@ +#!/usr/bin/bash + +CUDA_VISIBLE_DEVICES=6 /share/ymq/run/nvidia-asr/nvidia-asr.env/bin/python app/asr.py -p 9992 & +CUDA_VISIBLE_DEVICES=6 /share/ymq/run/nvidia-asr/nvidia-asr.env/bin/python app/asr.py -p 9992 & + diff --git a/stop.sh b/stop.sh new file mode 100644 index 0000000..4eb8841 --- /dev/null +++ b/stop.sh @@ -0,0 +1,3 @@ +#!/usr/bin/bash + +killname nvidia-asr.env