diff --git a/scraibe/diarisation.py b/scraibe/diarisation.py index c62bda0..570ac29 100644 --- a/scraibe/diarisation.py +++ b/scraibe/diarisation.py @@ -227,6 +227,8 @@ class Diariser: "Trying to find it nearby the config file.") pwd = model.split("/")[:-1] + pwd = "/".join(pwd) + path_to_model = os.path.join(pwd, "pytorch_model.bin") if not os.path.exists(path_to_model):