Spaces:
Running
Running
Add example.py code to the model using API
Browse files- app.py +1 -1
- example.py +31 -0
app.py
CHANGED
@@ -297,7 +297,7 @@ def inference(
|
|
297 |
if use_latest_take:
|
298 |
wav_gen = wav_gen_all[-1] # Get the latest generated wav
|
299 |
else:
|
300 |
-
wav_gen = torch.cat(wav_gen_all, dim=1)
|
301 |
|
302 |
duration = len(wav_gen[0]) / 44_100
|
303 |
torchaudio.save("audio.wav", wav_gen.cpu(), 44_100, encoding="PCM_S")
|
|
|
297 |
if use_latest_take:
|
298 |
wav_gen = wav_gen_all[-1] # Get the latest generated wav
|
299 |
else:
|
300 |
+
wav_gen = torch.cat(wav_gen_all, dim=1) # Concatenate all the generated wavs
|
301 |
|
302 |
duration = len(wav_gen[0]) / 44_100
|
303 |
torchaudio.save("audio.wav", wav_gen.cpu(), 44_100, encoding="PCM_S")
|
example.py
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
"""
|
2 |
+
pip3 install gradio_client
|
3 |
+
"""
|
4 |
+
|
5 |
+
from gradio_client import Client
|
6 |
+
|
7 |
+
client = Client("Yehor/radtts-uk-vocos-demo")
|
8 |
+
|
9 |
+
audio, metrics = client.predict(
|
10 |
+
param_0="Ви можете протестувати синтез мовлення українською мовою. Просто введіть текст, який ви хочете прослухати.", # text
|
11 |
+
param_1="Tetiana", # voice
|
12 |
+
param_2=1, # n_takes
|
13 |
+
param_3=False, # use_latest_take
|
14 |
+
param_4=1, # token_dur_scaling
|
15 |
+
param_5=0, # f0_mean
|
16 |
+
param_6=0, # f0_std
|
17 |
+
param_7=0, # energy_mean
|
18 |
+
param_8=0, # energy_std
|
19 |
+
param_9=0.8, # sigma_decoder
|
20 |
+
param_10=0.666, # sigma_token_duration
|
21 |
+
param_11=1, # sigma_f0
|
22 |
+
param_12=1, # sigma_energy
|
23 |
+
api_name="/inference_zerogpu",
|
24 |
+
)
|
25 |
+
|
26 |
+
filename = audio["value"]
|
27 |
+
print(filename)
|
28 |
+
|
29 |
+
print("---")
|
30 |
+
|
31 |
+
print(metrics)
|