Update app.py

app.py

@@ -7,11 +7,9 @@ import gradio as gr
 # --------------------------
 # TensorFlow config + Spleeter install
 # --------------------------
-# Enable eager execution & NumPy behavior for TF tensors
 tf.config.run_functions_eagerly(True)
 np_config.enable_numpy_behavior()
 
-# Install Spleeter manually if missing
 if not os.path.exists("/home/user/.spleeter_installed"):
     import install_spleeter
     open("/home/user/.spleeter_installed", "w").close()
@@ -20,31 +18,24 @@ if not os.path.exists("/home/user/.spleeter_installed"):
 # Import Spleeter
 # --------------------------
 from spleeter.separator import Separator
-from spleeter.audio.adapter import AudioAdapter
 
 # --------------------------
 # Initialize separator
 # --------------------------
 separator = Separator("spleeter:2stems", multiprocess=False)
-audio_loader = AudioAdapter.default()
 
 # --------------------------
 # Separation function
 # --------------------------
 def separate_audio(audio_file):
-    # Load audio as NumPy array
-    waveform, sample_rate = audio_loader.load(audio_file, sample_rate=44100)
-    
-    # Separate stems (vocals & accompaniment)
-    pred = separator.separate(waveform)
-    
-    # Save results to temporary folder
+    # Create temporary folder for outputs
     tmp = tempfile.mkdtemp()
-    vocals_path = os.path.join(tmp, "vocals.wav")
-    instr_path = os.path.join(tmp, "accompaniment.wav")
 
-    audio_loader.save(vocals_path, pred["vocals"], sample_rate)
-    audio_loader.save(instr_path, pred["accompaniment"], sample_rate)
+    # Separate directly to file (handles TensorFlow tensors internally)
+    separator.separate_to_file(audio_file, tmp, codec="wav")
+
+    vocals_path = os.path.join(tmp, os.path.splitext(os.path.basename(audio_file))[0], "vocals.wav")
+    instr_path = os.path.join(tmp, os.path.splitext(os.path.basename(audio_file))[0], "accompaniment.wav")
 
     return vocals_path, instr_path, "Done!"
 
@@ -58,4 +49,4 @@ iface = gr.Interface(
     title="Spleeter Vocal/Instrumental Separator"
 )
 
-iface.launch()
+iface.launch(share=True)  # Added public link