Mohssinibra commited on
Commit
a5cedb3
·
verified ·
1 Parent(s): f8308b9
Files changed (1) hide show
  1. app.py +3 -1
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="YOUR_HF_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):