Update app.py

app.py

@@ -29,14 +29,14 @@ def classify_audio(audio_file):
     wavs = torch.tensor(norm_wav).unsqueeze(0)
     
     pred = model(wavs, mask).detach().numpy()
+
+    output = ''
+#     for i, audio_pred in enumerate(pred):
+#         output[i] = {}
+    for att_i, att_val in enumerate(pred[0]):
+        output += "{}: \t{:0.4f}\n".format(id2label[att_i], att_val)
     
-    pred = {}
-    for i, audio_pred in enumerate(pred):
-        pred[i] = {}
-        for att_i, att_val in enumerate(audio_pred):
-            pred[i][id2label[att_i]] = att_val
-    
-    return pred
+    return output
 
 
 def main():
@@ -52,3 +52,4 @@ def main():
 if __name__ == '__main__':
     main()
 
+