Explorar o código

remove auxiliary audio extension (#1021)

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
Local State %!s(int64=2) %!d(string=hai) anos
pai
achega
c6e4e5efb3
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      whisper/utils.py

+ 1 - 0
whisper/utils.py

@@ -75,6 +75,7 @@ class ResultWriter:
 
     def __call__(self, result: dict, audio_path: str):
         audio_basename = os.path.basename(audio_path)
+        audio_basename = os.path.splitext(audio_basename)[0]
         output_path = os.path.join(
             self.output_dir, audio_basename + "." + self.extension
         )