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
|
# Initialize services
|
||||||
logger.info("Initializing services...")
|
logger.info("Initializing services...")
|
||||||
generation_service = GenerationService(
|
generation_service = GenerationService(
|
||||||
model_registry=model_registry,
|
registry=model_registry,
|
||||||
gpu_manager=gpu_manager,
|
gpu_manager=gpu_manager,
|
||||||
|
database=db,
|
||||||
output_dir=settings.output_dir,
|
output_dir=settings.output_dir,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ from pathlib import Path
|
|||||||
|
|
||||||
from src.ui.theme import get_custom_css
|
from src.ui.theme import get_custom_css
|
||||||
from src.ui.state import UIState, DEFAULT_PRESETS, PROMPT_SUGGESTIONS
|
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 (
|
from src.ui.tabs import (
|
||||||
create_dashboard_tab,
|
create_dashboard_tab,
|
||||||
create_musicgen_tab,
|
create_musicgen_tab,
|
||||||
@@ -193,15 +192,13 @@ class AudioCraftApp:
|
|||||||
# In Gradio 6.x, theme/css moved to launch()
|
# In Gradio 6.x, theme/css moved to launch()
|
||||||
# Create Blocks with no styling params
|
# Create Blocks with no styling params
|
||||||
with gr.Blocks() as app:
|
with gr.Blocks() as app:
|
||||||
# Header with VRAM monitor
|
# Header
|
||||||
with gr.Row():
|
with gr.Row():
|
||||||
with gr.Column(scale=4):
|
with gr.Column(scale=4):
|
||||||
gr.Markdown("# AudioCraft Studio")
|
gr.Markdown("# AudioCraft Studio")
|
||||||
with gr.Column(scale=1):
|
with gr.Column(scale=1):
|
||||||
vram_monitor = create_vram_monitor(
|
# Simple GPU status display
|
||||||
get_status_fn=self._get_gpu_status,
|
gpu_status = gr.Markdown("GPU: Loading...")
|
||||||
update_interval=5,
|
|
||||||
)
|
|
||||||
|
|
||||||
# Main tabs
|
# Main tabs
|
||||||
with gr.Tabs() as main_tabs:
|
with gr.Tabs() as main_tabs:
|
||||||
@@ -274,7 +271,7 @@ class AudioCraftApp:
|
|||||||
|
|
||||||
# Store component references
|
# Store component references
|
||||||
self.components = {
|
self.components = {
|
||||||
"vram_monitor": vram_monitor,
|
"gpu_status": gpu_status,
|
||||||
"dashboard": dashboard,
|
"dashboard": dashboard,
|
||||||
"musicgen": musicgen,
|
"musicgen": musicgen,
|
||||||
"audiogen": audiogen,
|
"audiogen": audiogen,
|
||||||
|
|||||||
Reference in New Issue
Block a user