Kabatubare commited on
Commit
780b961
·
verified ·
1 Parent(s): d75aa1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def predict_voice(audio_file_path):
42
  features = custom_feature_extraction(audio_file_path)
43
 
44
  with torch.no_grad():
45
- # Adjust the model prediction line if necessary to match your model's expected input
46
- outputs = model(inputs=features)
47
 
48
  logits = outputs.logits
49
  predicted_index = logits.argmax()
 
42
  features = custom_feature_extraction(audio_file_path)
43
 
44
  with torch.no_grad():
45
+ # Corrected: Directly pass the features tensor to the model
46
+ outputs = model(features)
47
 
48
  logits = outputs.logits
49
  predicted_index = logits.argmax()