Spaces:
Sleeping
Sleeping
Token
Browse files
app.py
CHANGED
@@ -7,6 +7,8 @@ from transformers import pipeline
|
|
7 |
import torchaudio
|
8 |
from pyannote.audio import Pipeline
|
9 |
|
|
|
|
|
10 |
# Charger le modèle de reconnaissance vocale
|
11 |
print("Chargement du modèle Wav2Vec2...")
|
12 |
stt_pipeline = pipeline("automatic-speech-recognition", model="boumehdi/wav2vec2-large-xlsr-moroccan-darija")
|
@@ -14,7 +16,7 @@ print("Modèle chargé avec succès !")
|
|
14 |
|
15 |
# Charger le pipeline de diarisation (détection des speakers)
|
16 |
print("Chargement du modèle de diarisation...")
|
17 |
-
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=
|
18 |
print("Modèle de diarisation chargé !")
|
19 |
|
20 |
def reduce_noise(audio, sr):
|
|
|
7 |
import torchaudio
|
8 |
from pyannote.audio import Pipeline
|
9 |
|
10 |
+
hf_token = os.getenv("diarizationToken")
|
11 |
+
|
12 |
# Charger le modèle de reconnaissance vocale
|
13 |
print("Chargement du modèle Wav2Vec2...")
|
14 |
stt_pipeline = pipeline("automatic-speech-recognition", model="boumehdi/wav2vec2-large-xlsr-moroccan-darija")
|
|
|
16 |
|
17 |
# Charger le pipeline de diarisation (détection des speakers)
|
18 |
print("Chargement du modèle de diarisation...")
|
19 |
+
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=hf_token)
|
20 |
print("Modèle de diarisation chargé !")
|
21 |
|
22 |
def reduce_noise(audio, sr):
|