Update functions.py

functions.py

@@ -167,6 +167,7 @@ def get_yt_audio(url):
 
     return audio_stream, title
 
+@st.cache_data
 def load_whisper_api(audio):
 
     '''Transcribe YT audio to text using Open AI API'''
@@ -226,8 +227,8 @@ def inference(link, upload, _asr_model):
                 transcriptions = []
                 
                 for i, chunk in enumerate(chunks):
-                    chunk.export(f'output/chunk_{i}.mp4', format='mp4')
-                    transcriptions.append(load_whisper_api(f'output/chunk_{i}.mp4')['text'])
+                    chunk.export(f'output/chunk_{i}_{link}.mp4', format='mp4')
+                    transcriptions.append(load_whisper_api(f'output/chunk_{i}_{link}.mp4')['text'])
     
                 results = ','.join(transcriptions)
 
@@ -264,8 +265,8 @@ def inference(link, upload, _asr_model):
                 st.info("`Transcribing uploaded audio...`")
                 
                 for i, chunk in enumerate(chunks):
-                    chunk.export(f'output/chunk_{i}.mp3', format='mp3')
-                    transcriptions.append(load_whisper_api('output/chunk_{i}.mp3')['text'])
+                    chunk.export(f'output/chunk_{i}.mp4', format='mp4')
+                    transcriptions.append(load_whisper_api('output/chunk_{i}.mp4')['text'])
     
                 results = ','.join(transcriptions)