update ui and add examples

2025-05-13 07:00:35 +00:00
parent 6db26bae5d
commit 6ca5bd880e
10 changed files with 258 additions and 119 deletions
@@ -71,6 +71,7 @@ def main(checkpoint_path, server_name, port, device_id, share, bf16, torch_compi
    demo = create_main_demo_ui(
        text2music_process_func=model_demo.__call__,
        sample_data_func=data_sampler.sample,
+        load_data_func=data_sampler.load_json,
    )
    demo.launch(server_name=server_name, server_port=port, share=share)