Update app.py

app.py

@@ -18,10 +18,10 @@ DEEPL_TOKEN = os.environ["DEEPL_TOKEN"]
 
 # Download video from Youtube
 def download_youtube_video(url):
- yt = YouTube(url)
- stream = yt.streams.filter(file_extension='mp4').first()
- output_path = stream.download()
- return output_path
+ yt = YouTube(url)
+ stream = yt.streams.filter(file_extension='mp4').first()
+ output_path = stream.download()
+ return output_path
 
 
 # Extract audio from video
@@ -54,7 +54,7 @@ def speech_diarization(audio_path, hf_token):
  import gc; gc.collect(); torch.cuda.empty_cache(); del model_a
 
  # 3. Assign speaker labels
- diarize_model = whisperx.DiarizationPipeline(model_name='pyannote/[email protected]', use_auth_token=hf_token, device=device)
+ diarize_model = whisperx.DiarizationPipeline(use_auth_token=hf_token, device=device)
 
  # add min/max number of speakers if known
  diarize_segments = diarize_model(audio)