Fix service init params and simplify VRAM monitor
- Fix GenerationService params: registry, database - Simplify VRAM display to basic Markdown for now 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
3
main.py
3
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,
|
||||
)
|
||||
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user