diff --git a/requirements.txt b/requirements.txt
index bed6304..8bc3e48 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ uvicorn[standard]>=0.24.0
 python-multipart>=0.0.6
 
 # GPU Monitoring
-pynvml>=11.5.0
+nvidia-ml-py>=12.535.0  # Replaces deprecated pynvml
 
 # Storage
 aiosqlite>=0.19.0
@@ -24,7 +24,7 @@ pydantic-settings>=2.1.0
 pyyaml>=6.0
 
 # Audio Processing
-numpy>=1.24.0
+numpy>=1.24.0,<2.0.0  # NumPy 2.x incompatible with current PyTorch
 scipy>=1.11.0
 librosa>=0.10.0
 soundfile>=0.12.0
diff --git a/src/ui/theme.py b/src/ui/theme.py
index 5b87905..acc4444 100644
--- a/src/ui/theme.py
+++ b/src/ui/theme.py
@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_theme() -> gr.themes.Base:
+def create_audiocraft_theme() -> gr.themes.Base:
     """Create custom theme for AudioCraft Studio.
 
     Returns:
@@ -301,3 +301,12 @@ CUSTOM_CSS = """
     }
 }
 """
+
+
+def get_custom_css() -> str:
+    """Get custom CSS for additional styling.
+
+    Returns:
+        CSS string
+    """
+    return CUSTOM_CSS