Spaces:

andrei-saceleanu
/

SSL_demo

Paused

Andrei-Iulian SĂCELEANU commited on Jun 10, 2023

Commit

1de0133

1 Parent(s): 1f3a9b6

fix preprocess

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ def ssl_predict2(audio_file, model_type):
     spectrogram = spectrogram.astype("float32")
     inputs = processor.preprocess(
-        np.repeat(spectrogram[:,:,:,np.newaxis],3,-1),
         image_mean=(-3.05,-3.05,-3.05),
         image_std=(2.33,2.33,2.33),
         return_tensors="tf"

     spectrogram = spectrogram.astype("float32")
     inputs = processor.preprocess(
+        np.repeat(spectrogram[np.newaxis,:,:,np.newaxis],3,-1),
         image_mean=(-3.05,-3.05,-3.05),
         image_std=(2.33,2.33,2.33),
         return_tensors="tf"