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