Update app.py

app.py

@@ -7,14 +7,14 @@ import tensorflow as tf
 from utils import _get_retina_bb, _pad_to_square
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def load_model(model_file):
     model = tf.keras.models.load_model(model_file, compile=False)
     print(f'Model {model_file} Loaded!')
     return model
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def load_gatekeeper():
     validator_model = tf.keras.models.load_model('checkpoints/ResNetV2-EyeQ-QA.tf')
     print('Gatekeeper Model Loaded!')