From 04bdcd7591e77390ddc46afefc4dd9180e274c3f Mon Sep 17 00:00:00 2001
From: Ftps <ftpsflandre@gmail.com>
Date: Tue, 26 Mar 2024 23:45:07 +0900
Subject: [PATCH] chore:detect inference models on fastapi

---
 rvc/wrapper/api/endpoints/inference.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/rvc/wrapper/api/endpoints/inference.py b/rvc/wrapper/api/endpoints/inference.py
index 86e169e..8ae0135 100644
--- a/rvc/wrapper/api/endpoints/inference.py
+++ b/rvc/wrapper/api/endpoints/inference.py
@@ -9,14 +9,26 @@ from pydantic import BaseModel
 from scipy.io import wavfile
 from base64 import b64encode
 from rvc.modules.vc.modules import VC
+import glob
+import os
 
 router = APIRouter()
+from dotenv import load_dotenv
+
+load_dotenv()
 
 
 @router.post("/inference")
 def inference(
-    modelpath: 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"]),
     sid: int = 0,
     f0_up_key: int = 0,