Update application file

app.py

@@ -22,6 +22,11 @@ async def lifespan(app: FastAPI):
     os.environ["TRANSFORMERS_CACHE"] = cache_dir
     os.environ["HF_HUB_CACHE"] = cache_dir
 
+    # Add numba cache directory for librosa
+    numba_cache_dir = os.path.join(cache_dir, "numba_cache")
+    os.makedirs(numba_cache_dir, exist_ok=True)
+    os.environ["NUMBA_CACHE_DIR"] = numba_cache_dir
+
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     torch_dtype = torch.float16 if device == "cuda:0" else torch.float32
     load_dotenv(override=True)  # Load environment variables from .env file