jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix reading form parameters at /inference endpoint

Browse Source
This commit is contained in:
Andrey Danilov
2024-05-17 16:57:04 +03:00
parent f9aee785a9
commit e4cb45016f
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
rvc/wrapper/api/endpoints/inference.py
View File

@@ -29,19 +29,19 @@ def inference(
for file in glob.glob(f"{os.getenv('weight_root')}/*") for file in glob.glob(f"{os.getenv('weight_root')}/*")
], ],
), ),
res_type: str = Query("blob", enum=["blob", "json"]), res_type: str = Form("blob", enum=["blob", "json"]),
sid: int = 0, sid: int = Form(0),
f0_up_key: int = 0, f0_up_key: int = Form(0),
f0_method: str = Query( f0_method: str = Form(
"rmvpe", enum=["pm", "harvest", "dio", "rmvpe", "rmvpe_gpu"] "rmvpe", enum=["pm", "harvest", "dio", "rmvpe", "rmvpe_gpu"]
), ),
f0_file: Path | None = None, f0_file: Path | None = None,
index_file: Path | None = None, index_file: Path | None = None,
index_rate: float = 0.75, index_rate: float = Form(0.75),
filter_radius: int = 3, filter_radius: int = Form(3),
resample_sr: int = 0, resample_sr: int = Form(0),
rms_mix_rate: float = 0.25, rms_mix_rate: float = Form(0.25),
protect: float = 0.33, protect: float = Form(0.33),
): ):
print(res_type) print(res_type)
vc = VC() vc = VC()