ckpt_update:modelCW-epoch=59-val_loss=1.69-val_recall=0.79

app.py

@@ -49,6 +49,9 @@ def pipeline(audio_file):
     
     with torch.no_grad():
         pred = torch.exp(classifier(processed_audio.unsqueeze(0)).squeeze())    # turning log_softmax into probabilities
+    
+    print({labels[i]: float(pred[i]) for i in range(3)})
+    print(classifier(processed_audio.unsqueeze(0)).squeeze())
 
     return {labels[i]: float(pred[i]) for i in range(3)}
 

model.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:abbf473f2a6445c3d4964c0fa0deb67a70d246149013d1bd42c7a64e60b9f8fe
-size 23819567
+oid sha256:37a14cfc80ee22924fa02c29ab51aadb62d986000ce043ccdcd34a285d35c1ec
+size 23819631