Update src/main.py

src/main.py

@@ -277,15 +277,15 @@ def vocal_only_pipeline(song_input, voice_model, pitch_change, is_webui=0,index_
                 orig_song_path, vocals_path = preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
         else:
             vocals_path = None
-            #for file in os.listdir(song_dir):
-            #    vocals_path = os.path.join(song_dir, file)
-            #return vocals_path
+            for file in os.listdir(song_dir):
+                vocals_path = os.path.join(song_dir, file)
+            return vocals_path
          
             #if any of the audio files aren't available or keep intermediate files, rerun preprocess
-            if any(path is None for path in paths):
-                orig_song_path, vocals_path= preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
-            else:
-                orig_song_path = paths
+            #if any(path is None for path in paths):
+            #    orig_song_path, vocals_path= preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
+            #else:
+            #    orig_song_path = paths
         #orig_song_path, vocals_path = preprocess_vocals_only(song_input, mdx_model_params, song_id, is_webui, input_type, progress)
         
         pitch_change = pitch_change * 12 + pitch_change_all