From 7c815afff46f2ea70ca54ba4f0d680a45543828f Mon Sep 17 00:00:00 2001 From: Andrey Danilov Date: Fri, 17 May 2024 16:58:04 +0300 Subject: [PATCH] Add logging for arguments of `/inference` endpoint --- rvc/wrapper/api/endpoints/inference.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/rvc/wrapper/api/endpoints/inference.py b/rvc/wrapper/api/endpoints/inference.py index 8719110..0c630b1 100644 --- a/rvc/wrapper/api/endpoints/inference.py +++ b/rvc/wrapper/api/endpoints/inference.py @@ -12,6 +12,10 @@ from rvc.modules.vc.modules import VC import glob import os +import logging + +logger: logging.Logger = logging.getLogger(__name__) + router = APIRouter() from dotenv import load_dotenv @@ -44,6 +48,8 @@ def inference( protect: float = Form(0.33), ): print(res_type) + logger.info(f"Arguments: {locals()}") + vc = VC() vc.get_vc(modelpath) tgt_sr, audio_opt, times, _ = vc.vc_inference(