Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import random
|
|
4 |
import numpy as np
|
5 |
import gradio as gr
|
6 |
import librosa
|
7 |
-
import spaces
|
8 |
from accelerate import Accelerator
|
9 |
from transformers import T5Tokenizer, T5EncoderModel
|
10 |
from diffusers import DDIMScheduler
|
@@ -62,7 +62,7 @@ autoencoder, unet, tokenizer, text_encoder, noise_scheduler, params = load_model
|
|
62 |
device)
|
63 |
|
64 |
|
65 |
-
@spaces.GPU
|
66 |
def generate_audio(text, length,
|
67 |
guidance_scale, guidance_rescale, ddim_steps, eta,
|
68 |
random_seed, randomize_seed):
|
@@ -95,7 +95,7 @@ def generate_audio(text, length,
|
|
95 |
return params['autoencoder']['sr'], pred
|
96 |
|
97 |
|
98 |
-
@spaces.GPU
|
99 |
def editing_audio(text, boundary,
|
100 |
gt_file, mask_start, mask_length,
|
101 |
guidance_scale, guidance_rescale, ddim_steps, eta,
|
|
|
4 |
import numpy as np
|
5 |
import gradio as gr
|
6 |
import librosa
|
7 |
+
# import spaces
|
8 |
from accelerate import Accelerator
|
9 |
from transformers import T5Tokenizer, T5EncoderModel
|
10 |
from diffusers import DDIMScheduler
|
|
|
62 |
device)
|
63 |
|
64 |
|
65 |
+
# @spaces.GPU
|
66 |
def generate_audio(text, length,
|
67 |
guidance_scale, guidance_rescale, ddim_steps, eta,
|
68 |
random_seed, randomize_seed):
|
|
|
95 |
return params['autoencoder']['sr'], pred
|
96 |
|
97 |
|
98 |
+
# @spaces.GPU
|
99 |
def editing_audio(text, boundary,
|
100 |
gt_file, mask_start, mask_length,
|
101 |
guidance_scale, guidance_rescale, ddim_steps, eta,
|