mirror of https://github.com/coqui-ai/TTS.git
Ping API
This commit is contained in:
parent
73d963718a
commit
8da3342676
|
@ -7,6 +7,7 @@ from pathlib import Path
|
||||||
from typing import Tuple
|
from typing import Tuple
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
import requests
|
||||||
from scipy.io import wavfile
|
from scipy.io import wavfile
|
||||||
|
|
||||||
from TTS.utils.audio.numpy_transforms import save_wav
|
from TTS.utils.audio.numpy_transforms import save_wav
|
||||||
|
@ -65,6 +66,11 @@ class CS_API:
|
||||||
self._speakers = None
|
self._speakers = None
|
||||||
self._check_token()
|
self._check_token()
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def ping_api():
|
||||||
|
URL = "https://coqui.gateway.scarf.sh/tts/api"
|
||||||
|
_ = requests.get(URL)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def speakers(self):
|
def speakers(self):
|
||||||
if self._speakers is None:
|
if self._speakers is None:
|
||||||
|
@ -80,6 +86,7 @@ class CS_API:
|
||||||
return ["Neutral", "Happy", "Sad", "Angry", "Dull"]
|
return ["Neutral", "Happy", "Sad", "Angry", "Dull"]
|
||||||
|
|
||||||
def _check_token(self):
|
def _check_token(self):
|
||||||
|
self.ping_api()
|
||||||
if self.api_token is None:
|
if self.api_token is None:
|
||||||
self.api_token = os.environ.get("COQUI_STUDIO_TOKEN")
|
self.api_token = os.environ.get("COQUI_STUDIO_TOKEN")
|
||||||
self.headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.api_token}"}
|
self.headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.api_token}"}
|
||||||
|
|
Loading…
Reference in New Issue