Fix file naming bug in inference server script

inference_server/server.py

@@ -9,7 +9,7 @@ sudo docker run --gpus all --runtime=nvidia --rm \
     -it rvc:v0
 curl -X GET http://localhost:8081/inference \
    -H 'Content-Type: application/json' \
-   -d '{"expName":"drake-200","audioFileName":"radwimps.mp3", "pitchShift": 0}'
+   -d '{"expName":"drake-20","audioFileName":"radwimps.mp3", "pitchShift": 0, "saveFileName":"radwimps_drake-20.wav"}'
 '''
 import json
 import os
@@ -37,15 +37,10 @@ def inference():
         exp_name = content['expName']
         audio_file_name = content['audioFileName']
         pitch_shift = content['pitchShift']
-
+        save_file_name = content['saveFileName']
+        
         model_path = exp_name + '.pth'
-        index_path_list = glob("/third_party/RVC/logs/{}/added_IVF*_Flat_nprobe_1.index".format(exp_name))
-        assert(len(index_path_list) == 1), "Index file not found"
-        index_path = index_path_list[0]
         audio_path = os.path.join('/inputDir', audio_file_name)
-        save_path = os.path.join('/outputDir', "{}_{}.wav".format(audio_file_name, exp_name))
-        error_log_path = os.path.join("/logs{}.log".format(exp_name))
-
         if not os.path.exists('/third_party/RVC/weights/{}'.format(model_path)):
             exception("Model doesn't exist")
             return json.dumps({"message":"Model doesn't exist"}), 404
@@ -53,6 +48,12 @@ def inference():
             exception("Audio file doesn't exist")
             return json.dumps({"message":"Audio file doesn't exist"}), 404
 
+        save_path = os.path.join('/outputDir', save_file_name)
+        error_log_path = os.path.join("/logs{}.log".format(exp_name))
+        index_path_list = glob("/third_party/RVC/logs/{}/added_IVF*_Flat_nprobe_1_v1.index".format(exp_name))
+        assert(len(index_path_list) == 1), "Index file not found"
+        index_path = index_path_list[0]
+        
         start_time = time.time()
         model_inference_single(model_path, index_path, audio_path, save_path, error_log_path, pitch_shift)
         end_time = time.time()