antonbol commited on
Commit
6ab7955
1 Parent(s): 2030d91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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])