Update transcription_diarization.py

transcription_diarization.py

@@ -15,7 +15,7 @@ def upload_to_s3(local_file_path, bucket_name, s3_file_key):
     s3_client.upload_file(local_file_path, bucket_name, s3_file_key)
     return f's3://{bucket_name}/{s3_file_key}'
 
-def transcribe_video(file_uri, job_name, max_speakers):
+def transcribe_video(file_uri, job_name):
     transcribe = boto3.client('transcribe',
                               aws_access_key_id=aws_access_key_id,
                               aws_secret_access_key=aws_secret_access_key,
@@ -27,8 +27,7 @@ def transcribe_video(file_uri, job_name, max_speakers):
         MediaFormat='mp4',
         IdentifyLanguage=True,
         Settings={
-            'ShowSpeakerLabels': True,
-            'MaxSpeakerLabels': max_speakers
+            'ShowSpeakerLabels': True
         }
     )
 
@@ -101,13 +100,13 @@ def extract_transcriptions_with_speakers(transcript_data):
     return transcriptions
 
 
-def diarize_audio(video_path, max_speakers):
+def diarize_audio(video_path):
     bucket_name = 'transcriptionjobbucket'
     s3_file_key = os.path.basename(video_path)
     file_uri = upload_to_s3(video_path, bucket_name, s3_file_key)
 
     job_name = f'transcription_job_{int(time.time())}'
-    transcript_url = transcribe_video(file_uri, job_name, max_speakers)
+    transcript_url = transcribe_video(file_uri, job_name)
 
     if transcript_url:
         transcript_data = download_transcript(transcript_url)