important bug fixed

app.py

@@ -28,15 +28,11 @@ model = load_model()
 def process_data(waveform_chunks):
     snore = 0
     other = 0
-    st.write("Reached stage 4")
     for chunk in waveform_chunks:
-        st.write("Reached stage 5")
-        input_tensor = torch.tensor(chunk).unsqueeze(0)
-        st.write("Reached stage 6")
-        st.write(input_tensor.shape)
+        input_tensor = torch.tensor(chunk).unsqueeze(0).to(torch.float32)
+        # st.write(input_tensor[0][98])
         result = model(input_tensor)
-        time.sleep(3)
-        st.write(result)
+        # st.write(result)
         if np.abs(result[0][0]) > np.abs(result[0][1]):
             other += 1
         else:
@@ -59,18 +55,11 @@ st.title('Real-Time Snore Detection App 😴')
 uploaded_file = st.file_uploader("Upload Sample", type=["wav"])
 if uploaded_file is not None:
     st.write("Analsysing...")
-    # time.sleep(2.5)
-    audio, sample_rate = librosa.load(uploaded_file, sr=None)
-    waveform = audio
-    # Set the chunk size
+    audio_bytes = uploaded_file.getvalue()
+    audio_array = np.frombuffer(audio_bytes, dtype=np.int16)
     chunk_size = 16000
-    st.write("Reached stage 2")
-    # Calculate the number of chunks
-    num_chunks = len(waveform) // chunk_size
-
-    # Reshape the waveform into chunks
-    waveform_chunks = np.array_split(waveform[:num_chunks * chunk_size], num_chunks)
-    st.write("Reached stage 3")
+    num_chunks = len(audio_array) // chunk_size
+    waveform_chunks = np.array_split(audio_array[:num_chunks * chunk_size], num_chunks)
     snore, other = process_data(waveform_chunks)
 
     total = snore + other
@@ -81,16 +70,16 @@ if uploaded_file is not None:
     percentages = [snore_percentage, other_percentage]
 
     st.write(f'Snore Percentage: {snore_percentage}')
-    # plt.figure(figsize=(8, 4))
-    # plt.barh(categories, percentages, color=['#ff0033', '#00ffee'])
-    # plt.xlabel('Percentage')
-    # plt.title('Percentage of Snoring')
-    # plt.xlim(0, 100)
-
-    # for i, percentage in enumerate(percentages):
-    #     plt.text(percentage, i, f' {percentage:.2f}%', va='center')
-    # st.write("DONE")
-    # st.pyplot(plt)
+    plt.figure(figsize=(8, 4))
+    plt.barh(categories, percentages, color=['#ff0033', '#00ffee'])
+    plt.xlabel('Percentage')
+    plt.title('Percentage of Snoring')
+    plt.xlim(0, 100)
+
+    for i, percentage in enumerate(percentages):
+        plt.text(percentage, i, f' {percentage:.2f}%', va='center')
+    st.write("DONE")
+    st.pyplot(plt)
     
 
     # # PERCENTAGE OF SNORING PLOT