Sirus1 commited on
Commit
5f1678c
1 Parent(s): 8913584

Update models/whisper_model.py

Browse files
Files changed (1) hide show
  1. models/whisper_model.py +1 -1
models/whisper_model.py CHANGED
@@ -14,7 +14,7 @@ class AudioTranslator():
14
 
15
  def __call__(self, video_path):
16
  print("Extract the audio results.")
17
- audio_results = self.model.transcribe(video_path)["segments"]
18
  print("Finished.")
19
  return audio_results
20
 
 
14
 
15
  def __call__(self, video_path):
16
  print("Extract the audio results.")
17
+ audio_results = self.model.transcribe(video_path,language="zh")["segments"]
18
  print("Finished.")
19
  return audio_results
20