Add default path to pyannote model with fallback option.

2024-04-19 17:36:34 +02:00
parent b075271b89
commit f7927fd524
2 changed files with 58 additions and 45 deletions
@@ -13,7 +13,8 @@ if CACHE_DIR != PYANNOTE_CACHE_DIR:

 WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper")
 PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote")
-PYANNOTE_DEFAULT_CONFIG = os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml") \
+PYANNOTE_DEFAULT_CONFIG = 'jaikinator/scraibe'
+PYANNOTE_FALLBACK_CONFIG = os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml") \
                            if os.path.exists(os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml")) \
                                else 'pyannote/speaker-diarization-3.1'