chong.zhang commited on
Commit
f630d6e
·
1 Parent(s): 9364cd5
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -89,7 +89,7 @@ def trim_audio(audio_file, cut_seconds=5):
89
  torchaudio.save(output_path, cutted_audio, sr)
90
  return output_path
91
 
92
- @spaces.GPU(duration=600)
93
  def music_generation(args):
94
  set_env_variables()
95
  model = InspireMusicUnified(
@@ -120,7 +120,7 @@ def music_generation(args):
120
  return output_path
121
 
122
 
123
- @spaces.GPU
124
  def demo_inspiremusic_t2m(text, model_name, chorus,
125
  output_sample_rate, max_generate_audio_seconds):
126
  args = get_args(
@@ -130,7 +130,7 @@ def demo_inspiremusic_t2m(text, model_name, chorus,
130
  max_generate_audio_seconds=max_generate_audio_seconds)
131
  return music_generation(args)
132
 
133
- @spaces.GPU
134
  def demo_inspiremusic_con(text, audio, model_name, chorus,
135
  output_sample_rate, max_generate_audio_seconds):
136
  args = get_args(
@@ -140,7 +140,7 @@ def demo_inspiremusic_con(text, audio, model_name, chorus,
140
  max_generate_audio_seconds=max_generate_audio_seconds)
141
  return music_generation(args)
142
 
143
- @spaces.GPU(duration=6000)
144
  def main():
145
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
146
  gr.Markdown("""
 
89
  torchaudio.save(output_path, cutted_audio, sr)
90
  return output_path
91
 
92
+ @spaces.GPU(duration=120)
93
  def music_generation(args):
94
  set_env_variables()
95
  model = InspireMusicUnified(
 
120
  return output_path
121
 
122
 
123
+ @spaces.GPU(duration=120)
124
  def demo_inspiremusic_t2m(text, model_name, chorus,
125
  output_sample_rate, max_generate_audio_seconds):
126
  args = get_args(
 
130
  max_generate_audio_seconds=max_generate_audio_seconds)
131
  return music_generation(args)
132
 
133
+ @spaces.GPU(duration=120)
134
  def demo_inspiremusic_con(text, audio, model_name, chorus,
135
  output_sample_rate, max_generate_audio_seconds):
136
  args = get_args(
 
140
  max_generate_audio_seconds=max_generate_audio_seconds)
141
  return music_generation(args)
142
 
143
+ @spaces.GPU(duration=120)
144
  def main():
145
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
146
  gr.Markdown("""