Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
9073612
1
Parent(s):
48b3a6b
update
Browse files
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("""
|