w

apis/zhtest.py

@@ -13,7 +13,7 @@ app = Celery('tasks', broker='redis://db.ptt.cx/0')
 @app.task
 def to_video(fname):
     os.chdir('/home/jared/to_video')
-    os.system('python3.7 save_mel.py --face nina-no-ges.mp4 --audio /var/wav/'+fname+'.wav --outfile /var/video/'+fname+'.mp4 --checkpoint_path gan.pth')
+    os.system('export CUDA_VISIBLE_DEVICES=0 ; python3.7 save_mel.py --face nina-no-ges.mp4 --audio /var/wav/'+fname+'.wav --outfile /var/video/'+fname+'.mp4 --checkpoint_path gan.pth')
     #text2mel_name="TACOTRON"
     return "ok"