nickmuchi commited on
Commit
6ce02f9
1 Parent(s): b4afe51

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +5 -3
functions.py CHANGED
@@ -195,6 +195,8 @@ def transcribe_yt_video(link, py_tube=True):
195
  transcriptions = []
196
 
197
  video_id = extract.video_id(link)
 
 
198
 
199
  for i, chunk in enumerate(chunks):
200
  chunk.export(f'output/chunk_{i}_{video_id}.mp4', format='mp4')
@@ -229,7 +231,7 @@ def transcribe_yt_video(link, py_tube=True):
229
 
230
  st.warning('File size larger than 24mb, applying chunking and transcription',icon="⚠️")
231
 
232
- song = AudioSegment.from_file(st.session_state['audio'], format='mp3')
233
 
234
  # PyDub handles time in milliseconds
235
  twenty_minutes = 20 * 60 * 1000
@@ -241,8 +243,8 @@ def transcribe_yt_video(link, py_tube=True):
241
  video_id = extract.video_id(link)
242
 
243
  for i, chunk in enumerate(chunks):
244
- chunk.export(f'output/chunk_{i}_{video_id}.mp3', format='mp3')
245
- transcriptions.append(load_whisper_api(f'output/chunk_{i}_{video_id}.mp3')['text'])
246
 
247
  results = ','.join(transcriptions)
248
 
 
195
  transcriptions = []
196
 
197
  video_id = extract.video_id(link)
198
+
199
+ ptint(video_id)
200
 
201
  for i, chunk in enumerate(chunks):
202
  chunk.export(f'output/chunk_{i}_{video_id}.mp4', format='mp4')
 
231
 
232
  st.warning('File size larger than 24mb, applying chunking and transcription',icon="⚠️")
233
 
234
+ song = AudioSegment.from_file(st.session_state['audio'], format='mp4')
235
 
236
  # PyDub handles time in milliseconds
237
  twenty_minutes = 20 * 60 * 1000
 
243
  video_id = extract.video_id(link)
244
 
245
  for i, chunk in enumerate(chunks):
246
+ chunk.export(f'output/chunk_{i}_{video_id}.mp3', format='mp4')
247
+ transcriptions.append(load_whisper_api(f'output/chunk_{i}_{video_id}.mp4')['text'])
248
 
249
  results = ','.join(transcriptions)
250