Spaces:
Sleeping
Sleeping
File size: 685 Bytes
627d3d7 02e90e4 627d3d7 02e90e4 627d3d7 02e90e4 627d3d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from modules.Enhancer.ResembleEnhance import reload_enhancer, unload_enhancer
from modules.api import utils as api_utils
from modules.api.Api import APIManager
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")
|