Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
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
|
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 |
-
|
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,
|