fixed bugs with hf tokens

2023-09-01 14:30:17 +02:00
parent d3e4c2dc75
commit f12c1396d6
2 changed files with 11 additions and 10 deletions
@@ -208,6 +208,8 @@ class Diariser:
        if not os.path.exists(model) and use_auth_token is None:
            use_auth_token = cls._get_token()
            model = 'pyannote/speaker-diarization'
+        elif not os.path.exists(model) and use_auth_token is not None:
+            model = 'pyannote/speaker-diarization'
            
        _model =  Pipeline.from_pretrained(model,
                                           use_auth_token = use_auth_token,