ChiBenevisamPas commited on
Commit
478d1da
·
verified ·
1 Parent(s): 67d7617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -21,10 +21,11 @@ def write_srt(transcription, output_file):
21
 
22
  def transcribe_video(video_file, language):
23
  # Transcribe the video to generate subtitles
24
- result = model.transcribe(video_file, language=language)
25
 
26
- # Save the transcription to an .srt file
27
- srt_file = "generated_subtitles.srt"
 
28
 
29
  # Write the transcription as subtitles
30
  write_srt(result, srt_file)
 
21
 
22
  def transcribe_video(video_file, language):
23
  # Transcribe the video to generate subtitles
24
+ result = model.transcribe(video_file.name, language=language)
25
 
26
+ # Get the video file name without extension and create the SRT file name
27
+ video_name = os.path.splitext(os.path.basename(video_file.name))[0]
28
+ srt_file = f"{video_name}.srt"
29
 
30
  # Write the transcription as subtitles
31
  write_srt(result, srt_file)