Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
|
2 |
import os
|
3 |
import random
|
4 |
import argparse
|
@@ -20,7 +20,7 @@ def generate_seed():
|
|
20 |
"value": new_seed
|
21 |
}
|
22 |
|
23 |
-
|
24 |
def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
|
25 |
|
26 |
torch.manual_seed(audio_seed_input)
|
|
|
1 |
+
import spaces
|
2 |
import os
|
3 |
import random
|
4 |
import argparse
|
|
|
20 |
"value": new_seed
|
21 |
}
|
22 |
|
23 |
+
@space.GPU
|
24 |
def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
|
25 |
|
26 |
torch.manual_seed(audio_seed_input)
|