Spaces:
Build error
Build error
Update model.py
Browse files
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
|
|
|
|
| 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()]
|