diff --git a/conf/config.json b/conf/config.json index c2dfc70..2640ecd 100644 --- a/conf/config.json +++ b/conf/config.json @@ -1,8 +1,8 @@ { "device" : "cuda", "asr_models":{ - "cn": "/share/ymq/models/nvidia/stt_zh_citrinet_1024_gamma_0_25/stt_zh_citrinet_1024_gamma_0_25.nemo", - "en": "/share/ymq/models/nvidia/stt_en_conformer_ctc_large/stt_en_conformer_ctc_large.nemo" + "cn": "/share/models/nvidia/stt_zh_citrinet_1024_gamma_0_25/stt_zh_citrinet_1024_gamma_0_25.nemo", + "en": "/share/models/nvidia/stt_en_conformer_ctc_large/stt_en_conformer_ctc_large.nemo" }, "logger":{ "name":"asr", diff --git a/nvidia-asr.service b/nvidia-asr.service index a566b8a..04d517f 100644 --- a/nvidia-asr.service +++ b/nvidia-asr.service @@ -4,6 +4,7 @@ Wants=systemd-networkd.service [Service] User=ymq Group=ymq +Type=forking WorkingDirectory=/share/ymq/run/nvidia-asr ExecStart=/share/ymq/run/nvidia-asr/start.sh ExecStop=/share/ymq/run/nvidia-asr/stop.sh diff --git a/start.sh b/start.sh old mode 100644 new mode 100755 index 18d6451..e22d1c1 --- a/start.sh +++ b/start.sh @@ -1,5 +1,7 @@ #!/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 & +rundir=/share/ymq/run/nvidia-asr +cd $rundir +CUDA_VISIBLE_DEVICES=6 $rundir/nvidia-asr.env/bin/python app/asr.py -p 9992 & +CUDA_VISIBLE_DEVICES=6 $rundir/nvidia-asr.env/bin/python app/asr.py -p 9992 & diff --git a/stop.sh b/stop.sh old mode 100644 new mode 100755 index 4eb8841..ee227ad --- a/stop.sh +++ b/stop.sh @@ -1,3 +1,4 @@ #!/usr/bin/bash -killname nvidia-asr.env +/d/ymq/bin/killname asr.py +echo "asr.py killed"