added cache

app.py

@@ -3,6 +3,14 @@ import streamlit as st
 from beluga import load_model, process_emotions, generate_prompt
 from emodeepface import check_image_rotation, process_photo
 
+# add streamlit cache to prevent multiple reloads of beluga model
+@st.cache(allow_output_mutation=True, suppress_st_warning=True)
+def load_cached_model():
+    return load_model()
+
+# begin loading beluga model and tokenizer
+model, tokenizer = load_cached_model()
+
 # begin loading beluga model and tokenizer
 model, tokenizer = load_model()