nachi1326 commited on
Commit
a03ff10
·
verified ·
1 Parent(s): 28cac0d

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +5 -1
model.py CHANGED
@@ -674,5 +674,9 @@ def pred_audio(path):
674
 
675
  # Classify based on combined probabilities
676
  combined_pred = (combined_prob >= 0.5).int()
 
 
 
 
677
 
678
- return combined_pred.item()
 
674
 
675
  # Classify based on combined probabilities
676
  combined_pred = (combined_prob >= 0.5).int()
677
+
678
+ cnn_pred = (cnn_prob >= 0.5).int()
679
+ lstm_pred = (lstm_prob >= 0.5).int()
680
+ dtdnn_pred = (dtdnn_prob >= 0.5).int()
681
 
682
+ return [cnn_pred.item(), lstm_pred.item(), dtdnn_pred.item()]