diff --git a/main.py b/main.py
index 0b00bb7..90aad08 100644
--- a/main.py
+++ b/main.py
@@ -59,8 +59,9 @@ async def initialize_services():
     # Initialize services
     logger.info("Initializing services...")
     generation_service = GenerationService(
-        model_registry=model_registry,
+        registry=model_registry,
         gpu_manager=gpu_manager,
+        database=db,
         output_dir=settings.output_dir,
     )
 
diff --git a/src/ui/app.py b/src/ui/app.py
index 449ac9f..4a53aac 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -7,7 +7,6 @@ from pathlib import Path
 
 from src.ui.theme import get_custom_css
 from src.ui.state import UIState, DEFAULT_PRESETS, PROMPT_SUGGESTIONS
-from src.ui.components.vram_monitor import create_vram_monitor
 from src.ui.tabs import (
     create_dashboard_tab,
     create_musicgen_tab,
@@ -193,15 +192,13 @@ class AudioCraftApp:
         # In Gradio 6.x, theme/css moved to launch()
         # Create Blocks with no styling params
         with gr.Blocks() as app:
-            # Header with VRAM monitor
+            # Header
             with gr.Row():
                 with gr.Column(scale=4):
                     gr.Markdown("# AudioCraft Studio")
                 with gr.Column(scale=1):
-                    vram_monitor = create_vram_monitor(
-                        get_status_fn=self._get_gpu_status,
-                        update_interval=5,
-                    )
+                    # Simple GPU status display
+                    gpu_status = gr.Markdown("GPU: Loading...")
 
             # Main tabs
             with gr.Tabs() as main_tabs:
@@ -274,7 +271,7 @@ class AudioCraftApp:
 
             # Store component references
             self.components = {
-                "vram_monitor": vram_monitor,
+                "gpu_status": gpu_status,
                 "dashboard": dashboard,
                 "musicgen": musicgen,
                 "audiogen": audiogen,