Kr08 commited on
Commit
99a4c60
·
verified ·
1 Parent(s): 4c0be85

Update chunkedTranscriber.py

Browse files
Files changed (1) hide show
  1. chunkedTranscriber.py +0 -3
chunkedTranscriber.py CHANGED
@@ -241,7 +241,6 @@ class ChunkedTranscriber:
241
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
242
 
243
  try:
244
- print(f"\n\n Chunk shape: {chunk_data['chunk'].shape}")
245
  # Language detection
246
  lid_processor, lid_model = self.load_lid_mms()
247
  lid_lang = self.language_identification(lid_model, lid_processor, chunk_data['chunk'])
@@ -346,7 +345,6 @@ class ChunkedTranscriber:
346
  'end_time': turn.end,
347
  'speaker': speaker
348
  })
349
- # print(f"\n\n Speaker Segments:\n{speaker_segments}\n")
350
 
351
  audio = self.load_audio(audio_path)
352
  chunks = self.preprocess_audio(audio)
@@ -366,7 +364,6 @@ class ChunkedTranscriber:
366
  translation_model,
367
  translation_tokenizer
368
  )
369
- print(f"\n\nResult:\n{result}")
370
  if result:
371
  for segment in speaker_segments:
372
  if int(segment['start_time']) <= int(chunk_data['start_time']) < int(segment['end_time']):
 
241
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
242
 
243
  try:
 
244
  # Language detection
245
  lid_processor, lid_model = self.load_lid_mms()
246
  lid_lang = self.language_identification(lid_model, lid_processor, chunk_data['chunk'])
 
345
  'end_time': turn.end,
346
  'speaker': speaker
347
  })
 
348
 
349
  audio = self.load_audio(audio_path)
350
  chunks = self.preprocess_audio(audio)
 
364
  translation_model,
365
  translation_tokenizer
366
  )
 
367
  if result:
368
  for segment in speaker_segments:
369
  if int(segment['start_time']) <= int(chunk_data['start_time']) < int(segment['end_time']):