Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+import logging
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
@@ -6,12 +8,21 @@ import torch
 from diffusers import AudioLDM2Pipeline
 from scipy.io.wavfile import write as write_wav
 import numpy as np
-import logging
 
 # --- Setup Logging ---
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# --- CRITICAL FIX for Hugging Face Spaces Permissions ---
+# Set the cache directory for all Hugging Face libraries BEFORE they are used.
+# This forces the model download and any temporary files to a writable location.
+cache_dir = "/tmp/huggingface_cache"
+os.environ["HF_HOME"] = cache_dir
+os.environ["HUGGINGFACE_HUB_CACHE"] = cache_dir
+# Create the directory if it doesn't exist, just in case.
+os.makedirs(cache_dir, exist_ok=True)
+logger.info(f"Hugging Face cache directory globally set to: {cache_dir}")
+
 # --- Initialize FastAPI App ---
 app = FastAPI()
 
@@ -24,12 +35,12 @@ 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, specifying a writable cache directory
-    # THIS IS THE FIX: Added cache_dir to point to a writable location
+    # The cache_dir argument is now redundant because of the environment variable,
+    # but we'll leave it for extra safety.
     pipeline = AudioLDM2Pipeline.from_pretrained(
         MODEL_REPO, 
         torch_dtype=torch_dtype,
-        cache_dir="/tmp/huggingface_cache"
+        cache_dir=cache_dir
     )
     pipeline = pipeline.to(device)
     logger.info("Model loaded successfully and moved to device.")