Update app.py

app.py

@@ -7,8 +7,6 @@ from joblib import memory
 
 memory = memory.Memory(location="cache", verbose=0)
 
-rvc = RVC.from_pretrained(MODEL_NAME)
-
 @memory.cache(ignore=["client"])
 def split(client, audio):
   result = client.predict(
@@ -18,7 +16,8 @@ def split(client, audio):
   )
   return result[0], result[1]
 
-def process_audio(client, audio, pitch_modification):
+def process_audio(client, model, audio, pitch_modification):
+  rvc = RVC.from_pretrained(model)
   vocal, bgm = split(client, audio)
   samples = rvc.convert(vocal, pitch_modification=pitch_modification)
 
@@ -50,7 +49,7 @@ with gr.Blocks() as demo:
 
     btn.submit(
         process_audio,
-        [client, gr.Audio(type="filepath"), gr.Slider(minimum=-36, maximum=36, value=0, step=1)],
+        [client, gr.Text("someone/model"), gr.Audio(type="filepath"), gr.Slider(minimum=-36, maximum=36, value=0, step=1)],
         [gr.Audio(label="Combined"), gr.Audio(label="Vocal"), gr.Audio(label="Background")]
     )