jhj0517 commited on
Commit
4c5ae77
·
1 Parent(s): 60697fb

Disable `safe_filename()` on local file

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -5
modules/whisper/whisper_base.py CHANGED
@@ -183,7 +183,6 @@ class WhisperBase(ABC):
183
  )
184
 
185
  file_name, file_ext = os.path.splitext(os.path.basename(file.name))
186
- file_name = safe_filename(file_name)
187
  subtitle, file_path = self.generate_and_write_file(
188
  file_name=file_name,
189
  transcribed_segments=transcribed_segments,
@@ -365,8 +364,8 @@ class WhisperBase(ABC):
365
  output_path: str
366
  output file path
367
  """
368
- timestamp = datetime.now().strftime("%m%d%H%M%S")
369
  if add_timestamp:
 
370
  output_path = os.path.join(output_dir, f"{file_name}-{timestamp}")
371
  else:
372
  output_path = os.path.join(output_dir, f"{file_name}")
@@ -374,17 +373,16 @@ class WhisperBase(ABC):
374
  if file_format == "SRT":
375
  content = get_srt(transcribed_segments)
376
  output_path += '.srt'
377
- write_file(content, output_path)
378
 
379
  elif file_format == "WebVTT":
380
  content = get_vtt(transcribed_segments)
381
  output_path += '.vtt'
382
- write_file(content, output_path)
383
 
384
  elif file_format == "txt":
385
  content = get_txt(transcribed_segments)
386
  output_path += '.txt'
387
- write_file(content, output_path)
 
388
  return content, output_path
389
 
390
  @staticmethod
 
183
  )
184
 
185
  file_name, file_ext = os.path.splitext(os.path.basename(file.name))
 
186
  subtitle, file_path = self.generate_and_write_file(
187
  file_name=file_name,
188
  transcribed_segments=transcribed_segments,
 
364
  output_path: str
365
  output file path
366
  """
 
367
  if add_timestamp:
368
+ timestamp = datetime.now().strftime("%m%d%H%M%S")
369
  output_path = os.path.join(output_dir, f"{file_name}-{timestamp}")
370
  else:
371
  output_path = os.path.join(output_dir, f"{file_name}")
 
373
  if file_format == "SRT":
374
  content = get_srt(transcribed_segments)
375
  output_path += '.srt'
 
376
 
377
  elif file_format == "WebVTT":
378
  content = get_vtt(transcribed_segments)
379
  output_path += '.vtt'
 
380
 
381
  elif file_format == "txt":
382
  content = get_txt(transcribed_segments)
383
  output_path += '.txt'
384
+
385
+ write_file(content, output_path)
386
  return content, output_path
387
 
388
  @staticmethod