Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ import typing as tp
|
|
17 |
import warnings
|
18 |
|
19 |
import gradio as gr
|
|
|
20 |
|
21 |
from audiocraft.data.audio import audio_write
|
22 |
from audiocraft.models import MAGNeT
|
@@ -129,7 +130,7 @@ def predict_batched(texts, melodies):
|
|
129 |
res = _do_predictions(texts, melodies)
|
130 |
return res
|
131 |
|
132 |
-
|
133 |
def predict_full(model, model_path, text, temperature, topp,
|
134 |
max_cfg_coef, min_cfg_coef,
|
135 |
decoding_steps1, decoding_steps2, decoding_steps3, decoding_steps4,
|
|
|
17 |
import warnings
|
18 |
|
19 |
import gradio as gr
|
20 |
+
import spaces
|
21 |
|
22 |
from audiocraft.data.audio import audio_write
|
23 |
from audiocraft.models import MAGNeT
|
|
|
130 |
res = _do_predictions(texts, melodies)
|
131 |
return res
|
132 |
|
133 |
+
@spaces.GPU(enable_queue=True)
|
134 |
def predict_full(model, model_path, text, temperature, topp,
|
135 |
max_cfg_coef, min_cfg_coef,
|
136 |
decoding_steps1, decoding_steps2, decoding_steps3, decoding_steps4,
|