fix

app.py

@@ -90,6 +90,9 @@ separator = Separator('spleeter:2stems')
 audio_loader = AudioAdapter.default()
 whisper_model = whisper.inference.load_model(os.path.join("whisper_pretrain", "medium.pt"))
 
+# warm up
+separator.separate_to_file('warm.wav', '/tmp/warm')
+
 
 def svc_change(argswave, argsspk):
     argsppg = "svc_tmp.ppg.npy"
@@ -183,17 +186,6 @@ def np_to_audio_segment(fp_arr):
     return sound
 
 
-def get_dtype_max_value(dtype):
-    if np.issubdtype(dtype, np.integer):
-        info = np.iinfo(dtype)
-        return info.max
-    elif np.issubdtype(dtype, np.floating):
-        info = np.finfo(dtype)
-        return info.max
-    else:
-        raise ValueError("不支持的 dtype 类型")
-
-
 def svc_main(sid, input_audio):
     if input_audio is None:
         return "You need to upload an audio", None
@@ -248,11 +240,13 @@ def auto_search(name):
         os.makedirs(save_music_path)
     config = {'logfilepath': 'musicdl.log', save_music_path: 'downloaded', 'search_size_per_source': 5, 'proxies': {}}
     albums = get_albums(keywords=name, config=config)
+    if len(albums) == 0:
+        return "No music found", None
     album = random.choice(albums)
     save_path = get_random_spit(album)
     fp = save_path
     signal, sampling_rate = soundfile.read(fp)
-    return sampling_rate, signal
+    return "Found a music", (sampling_rate, signal)
 
 
 app = gr.Blocks()
@@ -321,12 +315,13 @@ with app:
 
                 with gr.Column():
                     vc_input3 = gr.Audio(label="Upload Music Yourself")
-                    vc_search.click(auto_search, [vc_input2], [vc_input3])
                     vc_submit = gr.Button("Convert", variant="primary")
 
                 with gr.Column():
                     vc_output1 = gr.Textbox(label="Run Status")
                     vc_output2 = gr.Audio(label="Result Audio")
-                    vc_submit.click(svc_main, [sid, vc_input3], [vc_output1, vc_output2])
+
+                vc_search.click(auto_search, [vc_input2], [vc_output1, vc_input3])
+                vc_submit.click(svc_main, [sid, vc_input3], [vc_output1, vc_output2])
 
     app.launch()