Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import os
|
| 2 |
import random
|
| 3 |
import argparse
|
|
@@ -19,7 +20,7 @@ def generate_seed():
|
|
| 19 |
"value": new_seed
|
| 20 |
}
|
| 21 |
|
| 22 |
-
|
| 23 |
def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
|
| 24 |
|
| 25 |
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)
|