gpu zero

src/main.py

@@ -21,6 +21,8 @@ from pydub import AudioSegment
 from src.mdx import run_mdx, run_roformer
 from src.rvc import Config, load_hubert, get_vc, rvc_infer
 
+import spaces
+
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
 mdxnet_models_dir = os.path.join(BASE_DIR, 'mdxnet_models')
@@ -232,6 +234,7 @@ def combine_audio(audio_paths, output_path, main_gain, backup_gain, inst_gain, o
     instrumental_audio = AudioSegment.from_wav(audio_paths[2]) - 7 + inst_gain
     main_vocal_audio.overlay(backup_vocal_audio).overlay(instrumental_audio).export(output_path, format=output_format)
 
+@spaces.GPU
 def song_cover_pipeline(song_input, voice_model, pitch_change, keep_files,
                         is_webui=0, main_gain=0, backup_gain=0, inst_gain=0, index_rate=0.5, filter_radius=3,
                         rms_mix_rate=0.25, f0_method='rmvpe', crepe_hop_length=128, protect=0.33, pitch_change_all=0,