diff --git a/src/ui/app.py b/src/ui/app.py
index e8d0448..3c573c2 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -75,13 +75,11 @@ class AudioCraftApp:
     def _get_gpu_status(self) -> dict[str, Any]:
         """Get GPU memory status."""
         if self.gpu_manager:
-            return {
-                "used_gb": self.gpu_manager.get_used_memory() / 1024**3,
-                "total_gb": self.gpu_manager.total_memory / 1024**3,
-                "utilization_percent": self.gpu_manager.get_utilization(),
-                "available_gb": self.gpu_manager.get_available_memory() / 1024**3,
-            }
-        return {"used_gb": 0, "total_gb": 24, "utilization_percent": 0, "available_gb": 24}
+            try:
+                return self.gpu_manager.get_status()
+            except Exception:
+                pass
+        return {"total_gb": 24, "used_gb": 0, "free_gb": 24, "utilization_percent": 0, "available_for_models_gb": 14}
 
     async def _generate(self, **kwargs) -> tuple[Any, Any]:
         """Generate audio using the generation service."""