BrendaTellez commited on
Commit
d8895e4
·
1 Parent(s): 09af91b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -2,6 +2,9 @@ import streamlit as st
2
  from huggingface_hub import hf_hub_url, cached_download
3
  import numpy as np
4
  import tensorflow as tf
 
 
 
5
 
6
  REPO_ID = "BrendaTellez/sounds2"
7
  FILENAME = "RNN60-GRU.h5"
@@ -14,4 +17,16 @@ def download_model():
14
  return model
15
 
16
  model = download_model()
17
- st.write(model.summary)
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  from huggingface_hub import hf_hub_url, cached_download
3
  import numpy as np
4
  import tensorflow as tf
5
+ import librosa
6
+ from miniaudio import SampleFormat, decode
7
+
8
 
9
  REPO_ID = "BrendaTellez/sounds2"
10
  FILENAME = "RNN60-GRU.h5"
 
17
  return model
18
 
19
  model = download_model()
20
+
21
+ audio_bytes = st.file_uploader(
22
+ "Choose an audio (.wav) file", accept_multiple_files=False
23
+ )
24
+
25
+ decoded_audio = decode(
26
+ audio_bytes, nchannels=1, sample_rate=RATE, output_format=SampleFormat.SIGNED32
27
+ )
28
+ waveform = np.array(decoded_audio.samples)
29
+ MFCC = librosa.feature.mfcc(y=waveform, sr =RATE, hop_length = 512, n_mfcc = 60)
30
+
31
+ pred = model.predict(MFCC)
32
+ st.write(pred)