big update

app.py

@@ -25,6 +25,19 @@ s=0
 n=16000
 
 # Audio parameters
+def process_data(waveform_chunks):
+    st.write("Reached stage 4")
+    for chunk in waveform_chunks:
+        st.write("Reached stage 5")
+        input_tensor = torch.tensor(chunk).unsqueeze(0).to(torch.float32)
+        st.write("Reached stage 6")
+        result = model(input_tensor)
+        st.write("Reached stage 7")
+        if np.abs(result[0][0]) > np.abs(result[0][1]):
+            other += 1
+        else:
+            snore += 1
+    return snore, other
 
 st.sidebar.markdown(
     """
@@ -54,13 +67,7 @@ if uploaded_file is not None:
     # Reshape the waveform into chunks
     waveform_chunks = np.array_split(waveform[:num_chunks * chunk_size], num_chunks)
     st.write("Reached stage 3")
-    for chunk in waveform_chunks:
-        input_tensor = torch.tensor(chunk).unsqueeze(0).to(torch.float32)
-        result = model(input_tensor)
-        if np.abs(result[0][0]) > np.abs(result[0][1]):
-            other += 1
-        else:
-            snore += 1
+    snore, other = process_data(waveform_chunks)
 
     total = snore + other
     snore_percentage = (snore / total) * 100