19 lines
510 B
Python
19 lines
510 B
Python
from fastapi import APIRouter, Response, UploadFile, responses
|
|
|
|
from rvc.modules.uvr5.modules import UVR
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/inference")
|
|
def uvr(inputpath, outputpath, modelname, format):
|
|
uvr_module = UVR()
|
|
uvr_module.uvr_wrapper(
|
|
inputpath, outputpath, model_name=modelname, export_format=format
|
|
)
|
|
return responses.StreamingResponse(
|
|
audio,
|
|
media_type="audio/wav",
|
|
headers={"Content-Disposition": "attachment; filename=inference.wav"},
|
|
)
|