Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,9 @@ from pyannote.audio import Pipeline
|
|
15 |
warnings.filterwarnings("ignore")
|
16 |
logging.set_verbosity_error()
|
17 |
|
|
|
|
|
|
|
18 |
# Updated models by language
|
19 |
MODELS = {
|
20 |
"es": [
|
@@ -61,7 +64,7 @@ def detect_language(audio_path):
|
|
61 |
return max(langs, key=lambda x: x.prob).lang
|
62 |
|
63 |
def diarize_audio(wav_audio):
|
64 |
-
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
|
65 |
diarization = pipeline(wav_audio)
|
66 |
return diarization
|
67 |
|
|
|
15 |
warnings.filterwarnings("ignore")
|
16 |
logging.set_verbosity_error()
|
17 |
|
18 |
+
# Read the Hugging Face token from the environment variable
|
19 |
+
HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
|
20 |
+
|
21 |
# Updated models by language
|
22 |
MODELS = {
|
23 |
"es": [
|
|
|
64 |
return max(langs, key=lambda x: x.prob).lang
|
65 |
|
66 |
def diarize_audio(wav_audio):
|
67 |
+
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=HUGGINGFACE_TOKEN)
|
68 |
diarization = pipeline(wav_audio)
|
69 |
return diarization
|
70 |
|