Update app.py

app.py

@@ -17,7 +17,6 @@ app = FastAPI()
 
 # --- Model Loading ---
 # This section loads the AI model when the application starts.
-# This is crucial for performance, so it only happens once.
 MODEL_REPO = "cvssp/audioldm2"
 pipeline = None
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -25,8 +24,13 @@ torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
 try:
     logger.info(f"Attempting to load model '{MODEL_REPO}' on device: {device} with dtype: {torch_dtype}")
-    # Load the pre-trained AudioLDM2 pipeline
-    pipeline = AudioLDM2Pipeline.from_pretrained(MODEL_REPO, torch_dtype=torch_dtype)
+    # Load the pre-trained AudioLDM2 pipeline, specifying a writable cache directory
+    # THIS IS THE FIX: Added cache_dir to point to a writable location
+    pipeline = AudioLDM2Pipeline.from_pretrained(
+        MODEL_REPO, 
+        torch_dtype=torch_dtype,
+        cache_dir="/tmp/huggingface_cache"
+    )
     pipeline = pipeline.to(device)
     logger.info("Model loaded successfully and moved to device.")
 except Exception as e: