update

app.py

@@ -63,44 +63,44 @@ if upload_file is not None:
     # with open("output.wav", 'wb') as wav_file:
     #     wav_file.write(wav_make)
 
-sr, waveform = wavfile.read('saved_file.wav')
-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)
+    sr, waveform = wavfile.read('saved_file.wav')
+    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)