big update

app.py

@@ -17,24 +17,15 @@ import librosa
 
 # MODEL LOADING and INITIALISATION
 
-endReached = False
-
-s=0
-n=16000
-
-@st.cache_resource()
 def load_model():
     model = torch.jit.load("snorenetv1_small.ptl")
     model.eval()
     return model
-# model = load_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")
@@ -44,6 +35,7 @@ def process_data(waveform_chunks):
         st.write("Reached stage 6")
         st.write(input_tensor.shape)
         result = model(input_tensor)
+        time.sleep(3)
         st.write(result)
         if np.abs(result[0][0]) > np.abs(result[0][1]):
             other += 1