AbdullaShafeeg commited on
Commit
6c938cd
·
1 Parent(s): a15bed1

big update

Browse files
Files changed (2) hide show
  1. app.py +4 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -9,6 +9,8 @@ import torchaudio
9
  import wave
10
  import io
11
  from scipy.io import wavfile
 
 
12
  # MODEL LOADING and INITIALISATION
13
  model = torch.jit.load("snorenetv1_small.ptl")
14
  model.eval()
@@ -60,7 +62,8 @@ if upload_file is not None:
60
 
61
  # with open("output.wav", 'wb') as wav_file:
62
  # wav_file.write(wav_make)
63
-
 
64
  sr, waveform = wavfile.read(upload_file.name)
65
  snore = 0
66
  other = 0
 
9
  import wave
10
  import io
11
  from scipy.io import wavfile
12
+ import pydub
13
+
14
  # MODEL LOADING and INITIALISATION
15
  model = torch.jit.load("snorenetv1_small.ptl")
16
  model.eval()
 
62
 
63
  # with open("output.wav", 'wb') as wav_file:
64
  # wav_file.write(wav_make)
65
+ audio = pydub.AudioSegment.from_wav(upload_file)
66
+ audio.export(upload_file.name, format='wav')
67
  sr, waveform = wavfile.read(upload_file.name)
68
  snore = 0
69
  other = 0
requirements.txt CHANGED
@@ -4,4 +4,5 @@ matplotlib
4
  sounddevice
5
  torch
6
  torchaudio
7
- scipy
 
 
4
  sounddevice
5
  torch
6
  torchaudio
7
+ scipy
8
+ pydub