chong.zhang commited on
Commit
9073612
·
1 Parent(s): 48b3a6b
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- import spaces
3
  import gradio as gr
4
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
5
  import torchaudio
@@ -63,7 +63,7 @@ def trim_audio(audio_file, cut_seconds=5):
63
  torchaudio.save(output_path, cutted_audio, sr)
64
  return output_path
65
 
66
- @spaces.GPU
67
  def music_generation(args):
68
  set_env_variables()
69
  model = InspireMusicUnified(
@@ -94,7 +94,7 @@ def music_generation(args):
94
  return output_path
95
 
96
 
97
- @spaces.GPU
98
  def demo_inspiremusic_t2m(text, model_name, chorus,
99
  output_sample_rate, max_generate_audio_seconds):
100
  args = get_args(
@@ -104,7 +104,7 @@ def demo_inspiremusic_t2m(text, model_name, chorus,
104
  max_generate_audio_seconds=max_generate_audio_seconds)
105
  return music_generation(args)
106
 
107
- @spaces.GPU
108
  def demo_inspiremusic_con(text, audio, model_name, chorus,
109
  output_sample_rate, max_generate_audio_seconds):
110
  args = get_args(
@@ -114,7 +114,7 @@ def demo_inspiremusic_con(text, audio, model_name, chorus,
114
  max_generate_audio_seconds=max_generate_audio_seconds)
115
  return music_generation(args)
116
 
117
- @spaces.GPU
118
  def main():
119
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
120
  gr.Markdown("""
 
1
  import os
2
+ # import spaces
3
  import gradio as gr
4
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
5
  import torchaudio
 
63
  torchaudio.save(output_path, cutted_audio, sr)
64
  return output_path
65
 
66
+ # @spaces.GPU
67
  def music_generation(args):
68
  set_env_variables()
69
  model = InspireMusicUnified(
 
94
  return output_path
95
 
96
 
97
+ # @spaces.GPU
98
  def demo_inspiremusic_t2m(text, model_name, chorus,
99
  output_sample_rate, max_generate_audio_seconds):
100
  args = get_args(
 
104
  max_generate_audio_seconds=max_generate_audio_seconds)
105
  return music_generation(args)
106
 
107
+ # @spaces.GPU
108
  def demo_inspiremusic_con(text, audio, model_name, chorus,
109
  output_sample_rate, max_generate_audio_seconds):
110
  args = get_args(
 
114
  max_generate_audio_seconds=max_generate_audio_seconds)
115
  return music_generation(args)
116
 
117
+ # @spaces.GPU
118
  def main():
119
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
120
  gr.Markdown("""