Flux9665 commited on
Commit
aed09b0
·
verified ·
1 Parent(s): 5c20f98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -46,7 +46,7 @@ class ControllableInterface(torch.nn.Module):
46
  loudness_in_db
47
  ):
48
  if self.current_language != language:
49
- self.model = ToucanTTSInterface(device="cpu", tts_model_path="Meta", language=language)
50
  self.current_language = language
51
 
52
  self.wgan.set_latent(voice_seed)
@@ -59,11 +59,17 @@ class ControllableInterface(torch.nn.Module):
59
  embedding = self.wgan.modify_embed(controllability_vector)
60
  self.model.set_utterance_embedding(embedding=embedding)
61
 
 
 
62
  phones = self.model.text2phone.get_phone_string(prompt)
63
  if len(phones) > 1800:
64
- return
65
 
 
66
  print(prompt)
 
 
 
67
  wav, sr, fig = self.model(prompt,
68
  input_is_phones=False,
69
  duration_scaling_factor=duration_scaling_factor,
 
46
  loudness_in_db
47
  ):
48
  if self.current_language != language:
49
+ self.model.set_language(language)
50
  self.current_language = language
51
 
52
  self.wgan.set_latent(voice_seed)
 
59
  embedding = self.wgan.modify_embed(controllability_vector)
60
  self.model.set_utterance_embedding(embedding=embedding)
61
 
62
+ if len(prompt) > 1800:
63
+ raise AssertionError("The input is too long!")
64
  phones = self.model.text2phone.get_phone_string(prompt)
65
  if len(phones) > 1800:
66
+ raise AssertionError("The input is too long!")
67
 
68
+ print("\n\n")
69
  print(prompt)
70
+ print(language)
71
+ print("\n\n")
72
+
73
  wav, sr, fig = self.model(prompt,
74
  input_is_phones=False,
75
  duration_scaling_factor=duration_scaling_factor,