Yehor commited on
Commit
4bab821
·
1 Parent(s): d363231

Add example.py code to the model using API

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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) # 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")
 
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)