Upload 2 files

classifier.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:36b92beaea0d77a05bc45bf35c185cbcc576998b0967422db511d4fb251b37ea
-size 354369849
+oid sha256:2b8511ab7e9d0c0bf6c2a53de8906ad6876040cbf8b8c94cf9d6b565280c2f87
+size 354370070

config.pbtxt

@@ -6,15 +6,25 @@ input [
   {
     name: "audio_features"
     data_type: TYPE_FP32
-    dims: [80, 3000 ]  # [n_mels, time]
+    dims: [ 80, 3000 ]  # [n_mels, time]
   }
 ]
 
 output [
+  {
+    name: "logits"
+    data_type: TYPE_FP32
+    dims: [ 2 ]  # Binary classification
+  },
   {
     name: "probabilities"
     data_type: TYPE_FP32
     dims: [ 2 ]  # Binary classification
+  },
+  {
+    name: "predicted_class"
+    data_type: TYPE_INT64
+    dims: [ 1 ]  # Binary classification
   }
 ]