HyperMind-Labs commited on
Commit
2bc35ab
·
1 Parent(s): 218b02b
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -8,8 +8,8 @@ import torch
8
  import torchaudio
9
 
10
  # MODEL LOADING and INITIALISATION
11
- model = torch.jit.load("snorenetv1_small.ptl")
12
- model.eval()
13
 
14
  # Session state
15
  if 'text' not in st.session_state:
@@ -51,14 +51,14 @@ with st.expander('About this App'):
51
 
52
  wav_audio_data = st_audiorec()
53
  if wav_audio_data is not None:
54
- np_wav = np.frombuffer(wav_audio_data, dtype=np.int16)
55
- num_segments = len(np_wav)//100000
56
- segments_list = []
57
- for i in range(num_segments):
58
- start_sample = i * 100000
59
- end_sample = (i + 1) * 100000
60
- segment = np_wav[start_sample:end_sample]
61
- segments_list.append(segment)
62
 
63
  # wav_tensor = torch.tensor(np_wav, dtype=torch.float32)
64
  # segment_samples = int(44100 * 1)
@@ -69,7 +69,7 @@ if wav_audio_data is not None:
69
  # end_sample = (i + 1) * segment_samples
70
  # segment = wav_tensor[start_sample:end_sample]
71
  # segments_list.append(segment)
72
- st.write(segments_list)
73
  # PERCENTAGE OF SNORING PLOT
74
 
75
 
 
8
  import torchaudio
9
 
10
  # MODEL LOADING and INITIALISATION
11
+ # model = torch.jit.load("snorenetv1_small.ptl")
12
+ # model.eval()
13
 
14
  # Session state
15
  if 'text' not in st.session_state:
 
51
 
52
  wav_audio_data = st_audiorec()
53
  if wav_audio_data is not None:
54
+ # np_wav = np.frombuffer(wav_audio_data, dtype=np.int16)
55
+ # num_segments = len(np_wav)//100000
56
+ # segments_list = []
57
+ # for i in range(num_segments):
58
+ # start_sample = i * 100000
59
+ # end_sample = (i + 1) * 100000
60
+ # segment = np_wav[start_sample:end_sample]
61
+ # segments_list.append(segment)
62
 
63
  # wav_tensor = torch.tensor(np_wav, dtype=torch.float32)
64
  # segment_samples = int(44100 * 1)
 
69
  # end_sample = (i + 1) * segment_samples
70
  # segment = wav_tensor[start_sample:end_sample]
71
  # segments_list.append(segment)
72
+ st.write(type(wav_audio_data))
73
  # PERCENTAGE OF SNORING PLOT
74
 
75