Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
05598ad
1
Parent(s):
8ae3c92
update
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import os
|
| 2 |
-
import
|
| 3 |
import gradio as gr
|
| 4 |
from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
|
| 5 |
import torchaudio
|
|
@@ -60,7 +60,7 @@ def get_args(
|
|
| 60 |
print(args)
|
| 61 |
return args
|
| 62 |
|
| 63 |
-
|
| 64 |
def music_generation(args):
|
| 65 |
set_env_variables()
|
| 66 |
model = InspireMusicUnified(
|
|
@@ -109,6 +109,7 @@ def trim_audio(audio_file, cut_seconds=5):
|
|
| 109 |
torchaudio.save(output_path, cutted_audio, sr)
|
| 110 |
return output_path
|
| 111 |
|
|
|
|
| 112 |
def demo_inspiremusic_t2m(text, model_name, chorus,
|
| 113 |
output_sample_rate, max_generate_audio_seconds):
|
| 114 |
args = get_args(
|
|
@@ -118,6 +119,7 @@ def demo_inspiremusic_t2m(text, model_name, chorus,
|
|
| 118 |
max_generate_audio_seconds=max_generate_audio_seconds)
|
| 119 |
return music_generation(args)
|
| 120 |
|
|
|
|
| 121 |
def demo_inspiremusic_con(text, audio, model_name, chorus,
|
| 122 |
output_sample_rate, max_generate_audio_seconds):
|
| 123 |
args = get_args(
|
|
|
|
| 1 |
import os
|
| 2 |
+
import spaces
|
| 3 |
import gradio as gr
|
| 4 |
from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
|
| 5 |
import torchaudio
|
|
|
|
| 60 |
print(args)
|
| 61 |
return args
|
| 62 |
|
| 63 |
+
@spaces.GPU
|
| 64 |
def music_generation(args):
|
| 65 |
set_env_variables()
|
| 66 |
model = InspireMusicUnified(
|
|
|
|
| 109 |
torchaudio.save(output_path, cutted_audio, sr)
|
| 110 |
return output_path
|
| 111 |
|
| 112 |
+
@spaces.GPU
|
| 113 |
def demo_inspiremusic_t2m(text, model_name, chorus,
|
| 114 |
output_sample_rate, max_generate_audio_seconds):
|
| 115 |
args = get_args(
|
|
|
|
| 119 |
max_generate_audio_seconds=max_generate_audio_seconds)
|
| 120 |
return music_generation(args)
|
| 121 |
|
| 122 |
+
@spaces.GPU
|
| 123 |
def demo_inspiremusic_con(text, audio, model_name, chorus,
|
| 124 |
output_sample_rate, max_generate_audio_seconds):
|
| 125 |
args = get_args(
|