Spaces:

lenML
/

ChatTTS-Forge

Paused

ChatTTS-Forge / modules /api /impl /models_api.py

zhzluke96

update

d2b7e94 over 2 years ago

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")

	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")