fffiloni commited on
Commit
9ec41d3
1 Parent(s): 5973028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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,