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