Update app.py

app.py

@@ -13,6 +13,7 @@ from torch.cuda.amp import autocast
 import warnings
 import random
 from huggingface_hub import login
+import tempfile
 
 # Suppress warnings for cleaner output
 warnings.filterwarnings("ignore")
@@ -266,10 +267,11 @@ def generate_music(instrumental_prompt: str, cfg_scale: float, top_k: int, top_p
             if audio_chunk.shape[0] != 2:
                 raise ValueError(f"Expected stereo audio with shape (2, samples), got shape {audio_chunk.shape}")
 
-            temp_wav_path = f"temp_chunk_{i}.wav"
-            torchaudio.save(temp_wav_path, audio_chunk, sample_rate, bits_per_sample=24)
-            segment = AudioSegment.from_wav(temp_wav_path)
-            os.remove(temp_wav_path)
+            with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_wav:
+                temp_wav_path = temp_wav.name
+                torchaudio.save(temp_wav_path, audio_chunk, sample_rate, bits_per_sample=24)
+                segment = AudioSegment.from_wav(temp_wav_path)
+            os.unlink(temp_wav_path)
             audio_segments.append(segment)
 
             torch.cuda.empty_cache()
@@ -563,8 +565,6 @@ with gr.Blocks(css=css) as demo:
 
 # 9) TURN OFF OPENAPI/DOCS
 app = demo.launch(
-    server_name="0.0.0.0",
-    server_port=9999,
     share=False,
     inbrowser=False,
     show_error=True