Spaces:
Runtime error
Runtime error
File size: 1,823 Bytes
3b92d66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from starlette.responses import StreamingResponse
from tts import MelToWav, TextToMel
from advanced_tts import load_all_models, run_tts_paragraph
from typing import Optional
from pydantic import BaseModel
from fastapi import FastAPI, HTTPException
import uvicorn
import base64
import argparse
import json
import time
from argparse import Namespace
app = FastAPI()
class TextJson(BaseModel):
text: str
lang: Optional[str] = "hi"
noise_scale: Optional[float]=0.667
length_scale: Optional[float]=1.0
transliteration: Optional[int]=1
number_conversion: Optional[int]=1
split_sentences: Optional[int]=1
@app.post("/TTS/")
async def tts(input: TextJson):
text = input.text
lang = input.lang
args = Namespace(**input.dict())
args.wav = '../../results/api/'+str(int(time.time())) + '.wav'
if text:
sr, audio = run_tts_paragraph(args)
else:
raise HTTPException(status_code=400, detail={"error": "No text"})
## to return outpur as a file
audio = open(args.wav, mode='rb')
return StreamingResponse(audio, media_type="audio/wav")
# with open(args.wav, "rb") as audio_file:
# encoded_bytes = base64.b64encode(audio_file.read())
# encoded_string = encoded_bytes.decode()
# return {"encoding": "base64", "data": encoded_string, "sr": sr}
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-a", "--acoustic", required=True, type=str)
parser.add_argument("-v", "--vocoder", required=True, type=str)
parser.add_argument("-d", "--device", type=str, default="cpu")
parser.add_argument("-L", "--lang", type=str, required=True)
args = parser.parse_args()
load_all_models(args)
uvicorn.run(
"api:app", host="0.0.0.0", port=6006, log_level="debug"
)
|