samarth-ht commited on
Commit
dc4b0d6
·
verified ·
1 Parent(s): 2d36c70

Update scraibe/misc.py

Browse files
Files changed (1) hide show
  1. scraibe/misc.py +4 -4
scraibe/misc.py CHANGED
@@ -16,10 +16,10 @@ os.environ["PYANNOTE_CACHE"] = os.getenv(
16
 
17
  WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper")
18
  PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote")
19
- PYANNOTE_DEFAULT_CONFIG = ('pyannote/speaker-diarization-3.1','Jaikinator/ScrAIbe')
20
- # PYANNOTE_DEFAULT_CONFIG = os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml") \
21
- # if os.path.exists(os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml")) \
22
- # else ('Jaikinator/ScrAIbe', 'pyannote/speaker-diarization-3.1')
23
 
24
  SCRAIBE_TORCH_DEVICE = os.getenv("SCRAIBE_TORCH_DEVICE", "cuda" if is_available() else "cpu")
25
 
 
16
 
17
  WHISPER_DEFAULT_PATH = os.path.join(CACHE_DIR, "whisper")
18
  PYANNOTE_DEFAULT_PATH = os.path.join(CACHE_DIR, "pyannote")
19
+ # PYANNOTE_DEFAULT_CONFIG = ('pyannote/speaker-diarization-3.1','Jaikinator/ScrAIbe')
20
+ PYANNOTE_DEFAULT_CONFIG = os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml") \
21
+ if os.path.exists(os.path.join(PYANNOTE_DEFAULT_PATH, "config.yaml")) \
22
+ else ('Jaikinator/ScrAIbe', 'pyannote/speaker-diarization-3.1')
23
 
24
  SCRAIBE_TORCH_DEVICE = os.getenv("SCRAIBE_TORCH_DEVICE", "cuda" if is_available() else "cpu")
25