big update

app.py

@@ -22,10 +22,19 @@ endReached = False
 s=0
 n=16000
 
-# Audio parameters
-def process_data(waveform_chunks):
+@st.cache_resource()
+def load_model():
     model = torch.jit.load("snorenetv1_small.ptl")
     model.eval()
+    return model
+# model = load_model()
+
+if "model" not in st.session_state.keys():
+    st.session_state["model"] = load_model()
+model = st.session_state["model"]
+# Audio parameters
+def process_data(waveform_chunks):
+    
     snore = 0
     other = 0
     st.write("Reached stage 4")