push

app.py

@@ -1,15 +1,16 @@
 import streamlit as st
 import json
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def load_model():
     import adrd
     ckpt_path = './ckpt_densenet_emb_encoder_2_AUPR.pt'
     model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
     return model
 
+model = load_model()
+
 def predict_proba(data_dict):
-    model = load_model()
     pred_dict = model.predict_proba([data_dict])[1][0]
     return pred_dict