test1

app.py

@@ -4,7 +4,12 @@ import matplotlib.pyplot as plt
 import sounddevice as sd
 import numpy as np
 import pandas as pd
+import torch
+import torchaudio
 
+# MODEL LOADING and INITIALISATION
+model = torch.jit.load("snorenetv1_small.ptl")
+model.eval()
 
 # Session state
 if 'text' not in st.session_state:
@@ -47,38 +52,44 @@ with st.expander('About this App'):
 wav_audio_data = st_audiorec()
 if wav_audio_data is not None:
     np_wav = np.frombuffer(wav_audio_data, dtype=np.int16)
-
+    input_tensor = torch.tenosr(np_wav)
+    st.write(input_tensor.shape)
     # PERCENTAGE OF SNORING PLOT
 
 
-    model = np.random.rand(4, 5)
-
-    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)
+    
+    # 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)