Optimize launching directly instead of module

tools/api.py

@@ -9,6 +9,7 @@ from threading import Lock
 from typing import Annotated, Literal, Optional
 
 import librosa
+import pyrootutils
 import soundfile as sf
 import torch
 from kui.wsgi import (
@@ -25,6 +26,8 @@ from loguru import logger
 from pydantic import BaseModel
 from transformers import AutoTokenizer
 
+pyrootutils.setup_root(__file__, indicator=".project-root", pythonpath=True)
+
 from tools.llama.generate import launch_thread_safe_queue
 from tools.vqgan.inference import load_model as load_vqgan_model
 from tools.webui import inference

tools/webui.py

@@ -2,16 +2,18 @@ import gc
 import html
 import os
 import queue
-import threading
 from argparse import ArgumentParser
 from pathlib import Path
 
 import gradio as gr
 import librosa
+import pyrootutils
 import torch
 from loguru import logger
 from transformers import AutoTokenizer
 
+pyrootutils.setup_root(__file__, indicator=".project-root", pythonpath=True)
+
 from fish_speech.i18n import i18n
 from tools.llama.generate import launch_thread_safe_queue
 from tools.vqgan.inference import load_model as load_vqgan_model