Spaces:
Running
on
T4
Running
on
T4
try to figure out how ZeroGPU works
Browse files
app.py
CHANGED
@@ -117,6 +117,9 @@ class ControllableInterface(torch.nn.Module):
|
|
117 |
loudness_in_db=loudness_in_db)
|
118 |
return sr, wav, fig
|
119 |
|
|
|
|
|
|
|
120 |
|
121 |
title = "Controllable Text-to-Speech for over 7000 Languages"
|
122 |
article = "Check out the IMS Toucan TTS Toolkit at https://github.com/DigitalPhonetics/IMS-Toucan"
|
@@ -124,7 +127,7 @@ available_artificial_voices = 1000
|
|
124 |
path_to_iso_list = "Preprocessing/multilinguality/iso_to_fullname.json"
|
125 |
iso_to_name = load_json_from_path(path_to_iso_list)
|
126 |
text_selection = [f"{iso_to_name[iso_code]} Text ({iso_code})" for iso_code in iso_to_name]
|
127 |
-
gw =
|
128 |
controllable_ui = ControllableInterface(gan_wrapper=gw, available_artificial_voices=available_artificial_voices)
|
129 |
|
130 |
|
|
|
117 |
loudness_in_db=loudness_in_db)
|
118 |
return sr, wav, fig
|
119 |
|
120 |
+
@spaces.GPU
|
121 |
+
def get_gw():
|
122 |
+
return GanWrapper(os.path.join(MODELS_DIR, "Embedding", "embedding_gan.pt"), device="cuda" if torch.cuda.is_available() else "cpu")
|
123 |
|
124 |
title = "Controllable Text-to-Speech for over 7000 Languages"
|
125 |
article = "Check out the IMS Toucan TTS Toolkit at https://github.com/DigitalPhonetics/IMS-Toucan"
|
|
|
127 |
path_to_iso_list = "Preprocessing/multilinguality/iso_to_fullname.json"
|
128 |
iso_to_name = load_json_from_path(path_to_iso_list)
|
129 |
text_selection = [f"{iso_to_name[iso_code]} Text ({iso_code})" for iso_code in iso_to_name]
|
130 |
+
gw = get_gw()
|
131 |
controllable_ui = ControllableInterface(gan_wrapper=gw, available_artificial_voices=available_artificial_voices)
|
132 |
|
133 |
|