AbdullaShafeeg commited on
Commit
4db3eab
·
1 Parent(s): 32bfd0d

big update

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +2 -1
app.py CHANGED
@@ -13,6 +13,7 @@ import pydub
13
  import time
14
  import os
15
  import atexit
 
16
 
17
  # MODEL LOADING and INITIALISATION
18
  model = torch.jit.load("snorenetv1_small.ptl")
@@ -41,10 +42,9 @@ st.title('Real-Time Snore Detection App 😴')
41
  uploaded_file = st.file_uploader("Upload Sample", type=["wav"])
42
  if uploaded_file is not None:
43
  st.write("Analsysing...")
44
- time.sleep(7)
45
- audio = pydub.AudioSegment.from_wav(uploaded_file)
46
- audio_data = np.array(audio.get_array_of_samples())
47
- waveform= audio_data
48
  while(endReached == False):
49
  input_tensor = torch.tensor(waveform[s:n]).unsqueeze(0).to(torch.float32)
50
  result = model(input_tensor)
 
13
  import time
14
  import os
15
  import atexit
16
+ import librosa
17
 
18
  # MODEL LOADING and INITIALISATION
19
  model = torch.jit.load("snorenetv1_small.ptl")
 
42
  uploaded_file = st.file_uploader("Upload Sample", type=["wav"])
43
  if uploaded_file is not None:
44
  st.write("Analsysing...")
45
+ time.sleep(2.5)
46
+ audio, sample_rate = librosa.load(uploaded_file, sr=None)
47
+ waveform = audio
 
48
  while(endReached == False):
49
  input_tensor = torch.tensor(waveform[s:n]).unsqueeze(0).to(torch.float32)
50
  result = model(input_tensor)
requirements.txt CHANGED
@@ -5,4 +5,5 @@ sounddevice
5
  torch
6
  torchaudio
7
  scipy
8
- pydub
 
 
5
  torch
6
  torchaudio
7
  scipy
8
+ pydub
9
+ librosa