Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import subprocess
|
|
4 |
def vocal_remove(audio):
|
5 |
project = hopsworks.login()
|
6 |
mr = project.get_model_registry()
|
7 |
-
model = mr.get_best_model()
|
8 |
model_path = model.download()
|
9 |
model_path_pth = model_path + "/vocal_model.pth"
|
10 |
subprocess.run(["python", "inference.py", "--input", audio, "--pretrained_model", model_path_pth])
|
|
|
4 |
def vocal_remove(audio):
|
5 |
project = hopsworks.login()
|
6 |
mr = project.get_model_registry()
|
7 |
+
model = mr.get_best_model("vocal_remover", "validation_loss", "min")
|
8 |
model_path = model.download()
|
9 |
model_path_pth = model_path + "/vocal_model.pth"
|
10 |
subprocess.run(["python", "inference.py", "--input", audio, "--pretrained_model", model_path_pth])
|