Update functions.py

functions.py

@@ -119,19 +119,7 @@ def load_prompt():
 
 #     return audio_file, title
 
-
-@st.cache_data
-def get_yt_audio(url):
-
-    yt = YouTube(url)
-
-    # Get the first available audio stream and download it
-    audio_stream =  yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download()
-    title = audio_stream.split('\\')[-1].split('.')[0]
-
-    return audio_stream, title
-
-
+#load all required models and cache
 @st.cache_resource
 def load_models():
 
@@ -151,14 +139,22 @@ def load_models():
     
     return sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer, sbert
 
-@st.cache_resource
-def load_asr_model(asr_model_name):
-    asr_model = whisper.load_model(asr_model_name)
-    
-    return asr_model
+@st.cache_data
+def get_yt_audio(url):
+
+    '''Get YT video from given URL link'''
+    yt = YouTube(url)
+
+    # Get the first available audio stream and download it
+    audio_stream =  yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download()
+    title = audio_stream.split('\\')[-1].split('.')[0]
+
+    return audio_stream, title
 
 @st.cache_data
 def load_whisper_api(audio):
+
+    '''Transcribe YT audio to text using Open AI API'''
     file = open(audio, "rb")
     transcript = openai.Audio.translate("whisper-1", file)