Update pages/1_Earnings_Sentiment_Analysis_📈_.py

pages/1_Earnings_Sentiment_Analysis_📈_.py

@@ -6,9 +6,14 @@ from functions import *
 import validators
 
 #st.set_page_config(page_title="Earnings Sentiment Analysis", page_icon="📈")
+asr_model_options = ['base','small']
+asr_model_name = st.sidebar.selectbox("Whisper Model", options=asr_model_options, key='sbox')
 st.sidebar.header("Sentiment Analysis")
 st.markdown("## Earnings Sentiment Analysis with FinBert-Tone")
 
+#load whisper model
+asr_model = load_asr_model(asr_model_name)
+
 if "url" not in st.session_state:
     st.session_state.url = ''
 
@@ -16,7 +21,7 @@ try:
 
     if st.session_state['url'] is not None or st.session_state['upload'] is not None:
         
-        results, title = inference(st.session_state.url,st.session_state.upload)
+        results, title = inference(st.session_state.url,st.session_state.upload,asr_model)
     
         st.subheader(title)