ChatTTS-Forge / modules /api /impl /models_api.py
zhzluke96
update
d2b7e94
raw
history blame
685 Bytes
from modules.api import utils as api_utils
from modules.api.Api import APIManager
from modules.Enhancer.ResembleEnhance import reload_enhancer, unload_enhancer
from modules.models import reload_chat_tts, unload_chat_tts
def setup(app: APIManager):
@app.get("/v1/models/reload", response_model=api_utils.BaseResponse)
async def reload_models():
reload_chat_tts()
reload_enhancer()
return api_utils.success_response("Models reloaded")
@app.get("/v1/models/unload", response_model=api_utils.BaseResponse)
async def reload_models():
unload_chat_tts()
unload_enhancer()
return api_utils.success_response("Models unloaded")