Upload app.py with huggingface_hub

app.py

@@ -1,4 +1,5 @@
 import os
+import tempfile
 import torch
 import torchaudio
 import gradio as gr
@@ -42,6 +43,13 @@ def load_model(model_name):
     return MODEL_CACHE[model_name]
 
 
+# Pre-warm both models at startup so the first user request is fast
+print("Pre-warming models...")
+for _name in ["small-music", "small-sfx"]:
+    load_model(_name)
+print("All models ready!")
+
+
 def generate_audio(prompt, duration, steps, cfg_scale, seed, model_name):
     print(f"Generating with {model_name}: prompt='{prompt}', duration={duration}s, steps={steps}, cfg={cfg_scale}, seed={seed}")
 
@@ -60,7 +68,7 @@ def generate_audio(prompt, duration, steps, cfg_scale, seed, model_name):
     audio = rearrange(audio, "b d n -> d (b n)")
     audio = audio.to(torch.float32).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
 
-    output_path = "output.wav"
+    output_path = os.path.join(tempfile.gettempdir(), f"stable_audio_{seed}_{hash(prompt) & 0xFFFFFFFF:08x}.wav")
     torchaudio.save(output_path, audio, 44100)
     print("Generation complete!")
     return output_path