chenxie95 commited on
Commit
77b9459
·
1 Parent(s): 6634a6f
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  import torch
3
  import numpy as np
4
  import librosa
@@ -10,6 +11,9 @@ import tempfile
10
  import os
11
  from audio_controlnet.infer import AudioControlNet
12
 
 
 
 
13
  MAX_DURATION = 10.0 # seconds
14
 
15
  # -----------------------------
@@ -125,6 +129,7 @@ def save_temp_wav(audio):
125
  # -----------------------------
126
  # Generate audio
127
  # -----------------------------
 
128
  def generate_audio(text, cond_loudness, cond_pitch, cond_events):
129
  control = {}
130
  temp_files = []
 
1
  import gradio as gr
2
+ import spaces
3
  import torch
4
  import numpy as np
5
  import librosa
 
11
  import os
12
  from audio_controlnet.infer import AudioControlNet
13
 
14
+ import logging
15
+ logging.getLogger("gradio").setLevel(logging.WARNING)
16
+
17
  MAX_DURATION = 10.0 # seconds
18
 
19
  # -----------------------------
 
129
  # -----------------------------
130
  # Generate audio
131
  # -----------------------------
132
+ @spaces.GPU
133
  def generate_audio(text, cond_loudness, cond_pitch, cond_events):
134
  control = {}
135
  temp_files = []