Maximofn commited on
Commit
d5c8d04
1 Parent(s): eec6701

Import numpy at first

Browse files
Files changed (1) hide show
  1. translate_transcriptions.py +1 -0
translate_transcriptions.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
  from lang_list import LANGUAGE_NAME_TO_CODE, WHISPER_LANGUAGES
 
1
+ import numpy as np
2
  import torch
3
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
4
  from lang_list import LANGUAGE_NAME_TO_CODE, WHISPER_LANGUAGES