update

app.py

@@ -73,51 +73,46 @@ if wav_audio_data is not None:
     with open("output.wav", 'wb') as wav_file:
         wav_file.write(wav_make)
     sr, waveform = wavfile.read('output.wav')
-    input_tensor = torch.tensor(waveform[:16000]).unsqueeze(0).to(torch.float32)
-    st.write(input_tensor.shape)
-    result = model(input_tensor)
-    if np.abs(result[0][0]) > np.abs(result[0][1]):
-        st.write("NON_SNORING")
-    else:
-        st.write("SNORING")
-
+    snore = 0
+    other = 0
+    s=0
+    n=16000
+    endReached = False
+
+    while(endReached==False):
+        input_tensor = torch.tensor(waveform[s:n]).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
+        s += 16000
+        n += 16000
+        if(n >= len(waveform)):
+            endReached = True
     
     # PERCENTAGE OF SNORING PLOT
 
+    total = snore + other
+    snore_percentage = (snore / total) * 100
+    other_percentage = (other / total) * 100
+
+    categories = ["Snore", "Other"]
+    percentages = [snore_percentage, other_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.pyplot(plt)
+
 
+   
     
-    # waveform, sample_rate = torchaudio.load('test/0_10.wav')
-    # resampler = T.Resample(sample_rate, RESAMPLE_RATE, dtype=waveform.dtype)
-    # signal = resampler(waveform)
-    # signal = torch.mean(signal, dim=0, keepdim=True)
-    # ptl_model(signal)
-
-    # snore = 0
-    # other = 0
-
-    # for row in model:
-    #     for element in row:
-    #         if element > 0.5:
-    #             snore += 1
-    #         else:
-    #             other += 1
-
-    # total = snore + other
-    # snore_percentage = (snore / total) * 100
-    # other_percentage = (other / total) * 100
-
-    # categories = ["Snore", "Other"]
-    # percentages = [snore_percentage, other_percentage]
-
-    # plt.figure(figsize=(8, 4))
-    # plt.barh(categories, percentages, color=['#ff0033', '#00ffee'])
-    # plt.xlabel('Percentage')
-    # plt.title('Percentage of "Snore" and "Other"')
-    # plt.xlim(0, 100)
-
-    # for i, percentage in enumerate(percentages):
-    #     plt.text(percentage, i, f' {percentage:.2f}%', va='center')
-
-    # st.pyplot(plt)