Spaces:
Runtime error
Runtime error
AbdullaShafeeg
commited on
Commit
·
6c938cd
1
Parent(s):
a15bed1
big update
Browse files- app.py +4 -1
- 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
|