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: