deveix commited on
Commit
ee4841d
·
1 Parent(s): 6550579
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -212,7 +212,7 @@ SAMPLE_RATE = 44100 # sample rate to use
212
  N_MELS = 128 # number of Mel bands to generate
213
  HOP_LENGTH = 512 # number of samples between successive frames
214
 
215
- def preprocess_audio(file_path):
216
  try:
217
  # Load the audio file
218
  audio, sr = librosa.load(file_path, sr=SAMPLE_RATE)
@@ -289,7 +289,7 @@ async def handle_cnn(file: UploadFile = File(...)):
289
 
290
  spectrograms = []
291
 
292
- clips = preprocess_audio(temp_filename)
293
  for clip in clips:
294
  spectrogram = generate_spectrogram(clip)
295
  if np.isnan(spectrogram).any() or np.isinf(spectrogram).any():
 
212
  N_MELS = 128 # number of Mel bands to generate
213
  HOP_LENGTH = 512 # number of samples between successive frames
214
 
215
+ def preprocess_audio_cnn(file_path):
216
  try:
217
  # Load the audio file
218
  audio, sr = librosa.load(file_path, sr=SAMPLE_RATE)
 
289
 
290
  spectrograms = []
291
 
292
+ clips = preprocess_audio_cnn(temp_filename)
293
  for clip in clips:
294
  spectrogram = generate_spectrogram(clip)
295
  if np.isnan(spectrogram).any() or np.isinf(spectrogram).any():