IvanLayer7 commited on
Commit
8448657
·
1 Parent(s): 7118df6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -9,7 +9,6 @@ import librosa
9
  import librosa.display
10
 
11
  def transform_data(audio):
12
- print("audio: ", audio)
13
  # Lets transform the dataset so we can apply the predictions
14
  X, sample_rate = librosa.load(audio
15
  ,res_type='kaiser_fast'
@@ -17,13 +16,10 @@ def transform_data(audio):
17
  ,sr=44100
18
  ,offset=0.5
19
  )
20
-
21
- print("HEREEEE", sample_rate, X.shape)
22
-
23
  sample_rate = np.array(sample_rate)
24
- mfccs = np.mean(librosa.feature.mfcc(y=X, sr=sample_rate, n_mfcc=13),axis=0)
25
- newdf = pd.DataFrame(data=mfccs).T
26
- return newdf
27
 
28
  def predict(newdf, loaded_model):
29
  # Apply predictions
 
9
  import librosa.display
10
 
11
  def transform_data(audio):
 
12
  # Lets transform the dataset so we can apply the predictions
13
  X, sample_rate = librosa.load(audio
14
  ,res_type='kaiser_fast'
 
16
  ,sr=44100
17
  ,offset=0.5
18
  )
 
 
 
19
  sample_rate = np.array(sample_rate)
20
+ mfccs = np.mean(librosa.feature.mfcc(y=X, sr=sample_rate, n_mfcc=30),axis=0)
21
+ mfccs = np.expand_dims(mfccs, axis=-1)
22
+ return mfccs
23
 
24
  def predict(newdf, loaded_model):
25
  # Apply predictions