ww

apis/main.py

@@ -11,6 +11,7 @@ from typing import Optional
 from fastapi import FastAPI
 from pydantic import BaseModel
 import rpyc
+import zhtest
 
 
 class TTS(BaseModel):
@@ -26,6 +27,8 @@ app = FastAPI()
 async def run_tts(item: TTS):
     print(item.txt)
     print(item.num)
+    fname=zhtest.to_wav.delay(item.txt)
+    print(fname)
     return {"OK": "200"}
 
 

apis/zhtest.py

@@ -0,0 +1,21 @@
+import zhtts
+import celery
+import string
+import random
+# export CUDA_VISIBLE_DEVICES=0 ; python3.7 train.py -content /tmp/demo.wav -style input/girl52.wav
+#py -3.7 save_mel.py --face a1.mp4 --audio eng1.mp3 --checkpoint_path wav2lip_gan.pth
+#export CUDA_VISIBLE_DEVICES=0 ; python3.7 save_mel.py --face nina.mp4 --audio /tmp/demo.wav --checkpoint_path wav2lip_gan.pth
+from celery import Celery
+
+app = Celery('tasks', broker='redis://db.ptt.cx/0')
+
+@app.task
+def to_wav(txt):
+    letters = string.digits
+    rstr= ''.join(random.choice(letters) for i in range(10))
+    #text2mel_name="TACOTRON"
+    tts = zhtts.TTS() # use fastspeech2 by default
+    #tts = zhtts.TTS(text2mel_name="TACOTRON") # use fastspeech2 by default
+
+    tts.text2wav(txt, "/var/wav/"+rstr+".wav")
+    return rstr