Update app.py

app.py

@@ -1,6 +1,4 @@
 import os
-import tempfile
-from pathlib import Path
 
 import gradio as gr
 import numpy as np
@@ -11,20 +9,19 @@ from huggingface_hub import login
 from stable_audio_3 import StableAudioModel
 
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
+hf_token = os.environ.get("HF_TOKEN")
 
-if HF_TOKEN:
-    login(token=HF_TOKEN)
+if not hf_token:
+    raise RuntimeError("HF_TOKEN Space secret is missing.")
 
+login(token=hf_token)
 
-# ZeroGPU emulates CUDA during Space startup.
-# Loading and moving the model here is more efficient than doing it
-# inside the @spaces.GPU function.
-model = StableAudioModel.from_pretrained("small-sfx")
-model = model.to("cuda")
+model = StableAudioModel.from_pretrained(
+    "small-sfx",
+    device="cuda",
+)
 model.eval()
 
-
 def requested_gpu_duration(prompt: str, duration: float) -> int:
     """
     Reserve a modest amount of GPU time.