Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ from diffusers import DiffusionPipeline
|
|
10 |
from huggingface_hub import login
|
11 |
import numpy as np
|
12 |
import spaces
|
|
|
13 |
|
14 |
@spaces.GPU
|
15 |
def guessanImage(model, image):
|
@@ -27,12 +28,14 @@ def guessanAge(model, image):
|
|
27 |
|
28 |
@spaces.GPU(duration=120)
|
29 |
def text2speech(text, no, sample):
|
|
|
30 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
31 |
if sample is None:
|
32 |
sample = "sampleaudio/abraham.wav"
|
33 |
if len(text) > 0:
|
|
|
34 |
tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2")
|
35 |
-
wav = tts.tts_to_file(text=text, file_path="output.wav", speaker_wav=sample, language="en")
|
36 |
return wav
|
37 |
|
38 |
@spaces.GPU
|
|
|
10 |
from huggingface_hub import login
|
11 |
import numpy as np
|
12 |
import spaces
|
13 |
+
import time
|
14 |
|
15 |
@spaces.GPU
|
16 |
def guessanImage(model, image):
|
|
|
28 |
|
29 |
@spaces.GPU(duration=120)
|
30 |
def text2speech(text, no, sample):
|
31 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
32 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
33 |
if sample is None:
|
34 |
sample = "sampleaudio/abraham.wav"
|
35 |
if len(text) > 0:
|
36 |
+
epoch_time = int(time.time())
|
37 |
tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2")
|
38 |
+
wav = tts.tts_to_file(text=text, file_path="output-"+epoch_time+".wav", speaker_wav=sample, language="en").to(device)
|
39 |
return wav
|
40 |
|
41 |
@spaces.GPU
|