Update functions.py

functions.py

@@ -83,6 +83,14 @@ def load_models():
     
     return sent_pipe, sum_pipe, ner_pipe, cross_encoder, sbert
 
+@st.cache_data
+def load_asr_model(model_name):
+
+    '''Load the open source  whisper model in cases where the API is not working'''
+    model = whisper.load_model(model_name)
+
+    return model
+
 @st.cache_resource
 def get_spacy():
     nlp = en_core_web_lg.load()
@@ -134,13 +142,6 @@ def get_yt_audio_dl(url):
 
     return audio_file, title
 
-@st.cache_data
-def load_asr_model(model_name):
-
-    '''Load the open source  whisper model in cases where the API is not working'''
-    model = whisper.load_model(model_name)
-
-    return model
     
 @st.cache_data
 def load_whisper_api(audio):
@@ -245,7 +246,7 @@ def transcribe_yt_video(url, py_tube=True):
     return results, title
 
 @st.cache_data
-def inference(link, upload, _asr_model):
+def inference(link, upload):
     '''Convert Youtube video or Audio upload to text'''
     
     try: