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

chore:detect inference models on fastapi

Browse Source
This commit is contained in:
Ftps
2024-03-26 23:45:07 +09:00
parent 68f5d08b25
commit 04bdcd7591
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,14 +9,26 @@ from pydantic import BaseModel
from scipy.io import wavfile from scipy.io import wavfile
from base64 import b64encode from base64 import b64encode
from rvc.modules.vc.modules import VC from rvc.modules.vc.modules import VC
import glob
import os
router = APIRouter() router = APIRouter()
from dotenv import load_dotenv
load_dotenv()
@router.post("/inference") @router.post("/inference")
def inference( def inference(
modelpath: Path | UploadFile,
input_audio: Path | UploadFile, input_audio: Path | UploadFile,
modelpath: Path
| UploadFile = Body(
...,
enum=[
os.path.basename(file)
for file in glob.glob(f"{os.getenv('weight_root')}/*")
],
),
res_type: str = Query("blob", enum=["blob", "json"]), res_type: str = Query("blob", enum=["blob", "json"]),
sid: int = 0, sid: int = 0,
f0_up_key: int = 0, f0_up_key: int = 0,