fix: Bug fixed - earlier only 6 was getting predicted every time

app.py

@@ -19,7 +19,8 @@ def mel_spectrogram_tfm(file):
 
 def classify(audio):
     spec_db = mel_spectrogram_tfm(audio)
-    img = PILImage.create(spec_db)
+    img = 'tmp.png'
+    plt.imsave(img, spec_db)
     pred, idx, prob = learn.predict(img)
     return dict(zip(categories, map(float, prob)))