Kr08 commited on
Commit
5812712
·
verified ·
1 Parent(s): ea6ce06

Update chunkedTranscriber.py

Browse files
Files changed (1) hide show
  1. chunkedTranscriber.py +8 -1
chunkedTranscriber.py CHANGED
@@ -373,9 +373,16 @@ class ChunkedTranscriber:
373
  # results.append(result)
374
 
375
  # Merge close segments and clean up
 
376
  merged_results = self.merge_close_segments(results)
377
 
378
- return merged_results
 
 
 
 
 
 
379
 
380
 
381
  def load_audio(self, audio_path):
 
373
  # results.append(result)
374
 
375
  # Merge close segments and clean up
376
+
377
  merged_results = self.merge_close_segments(results)
378
 
379
+ _translation = ""
380
+ _output = ""
381
+ for res in merged_results:
382
+ _translation+=res['translated']
383
+ _output+=f"{res['start_time']}-{res['end-time']} - Speaker: {res['speaker'].split('_')[1]} - Language: {res['detected_language']}\n Text: {res['text']}\n Translation: {res['translated']}\n\n"
384
+ logger.info(f"\n\n TRANSLATION: {_translation}")
385
+ return _output
386
 
387
 
388
  def load_audio(self, audio_path):