fix: gpu header display

2025-11-27 15:29:16 +01:00
parent b80defd929
commit 2f9bdb122f
1 changed files with 9 additions and 0 deletions
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -283,6 +283,15 @@ class AudioCraftApp:
                "settings": settings,
            }

+            # Update header GPU status on page load
+            def update_header_gpu():
+                status = self._get_gpu_status()
+                used = status.get('used_gb', 0)
+                total = status.get('total_gb', 24)
+                return f"GPU: {used:.1f}/{total:.1f} GB"
+
+            app.load(fn=update_header_gpu, outputs=[gpu_status])
+
        self.app = app
        return app