reab5555 commited on
Commit
cd4a01b
·
verified ·
1 Parent(s): 9642c0e

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +2 -2
video_processing.py CHANGED
@@ -156,8 +156,8 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
156
  progress(0.8, "Extracting audio and performing voice analysis")
157
  audio_path = extract_audio_from_video(video_path)
158
  diarization = diarize_speakers(audio_path)
159
- voice_model = Model.from_pretrained("pyannote/embedding")
160
- voice_embeddings = get_speaker_embeddings(audio_path, diarization, voice_model)
161
 
162
  progress(0.85, "Performing anomaly detection")
163
  embedding_columns = [col for col in df.columns if col.startswith('Raw_Embedding_')]
 
156
  progress(0.8, "Extracting audio and performing voice analysis")
157
  audio_path = extract_audio_from_video(video_path)
158
  diarization = diarize_speakers(audio_path)
159
+ voice_embeddings = get_speaker_embeddings(audio_path, diarization, "pyannote/embedding")
160
+
161
 
162
  progress(0.85, "Performing anomaly detection")
163
  embedding_columns = [col for col in df.columns if col.startswith('Raw_Embedding_')]