jgoot commited on
Commit
e1ad392
·
verified ·
1 Parent(s): d477208

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -46,6 +46,9 @@ def predict_age_gender(audio_input):
46
  if signal.ndim > 1:
47
  signal = signal.mean(axis=1)
48
 
 
 
 
49
  # Run inference
50
  output = model(signal, sampling_rate)
51
 
@@ -64,9 +67,9 @@ def predict_age_gender(audio_input):
64
  **Predicted Gender**: {predicted_gender.capitalize()}
65
 
66
  **Gender Confidence Scores**:
67
- - Female: {gender_scores[0]:.2%}
68
- - Male: {gender_scores[1]:.2%}
69
- - Child: {gender_scores[2]:.2%}
70
  """
71
 
72
  return result
 
46
  if signal.ndim > 1:
47
  signal = signal.mean(axis=1)
48
 
49
+ # Convert to float32 (ONNX requires this)
50
+ signal = signal.astype(np.float32)
51
+
52
  # Run inference
53
  output = model(signal, sampling_rate)
54
 
 
67
  **Predicted Gender**: {predicted_gender.capitalize()}
68
 
69
  **Gender Confidence Scores**:
70
+ - Female: {gender_scores[0]:.2f}
71
+ - Male: {gender_scores[1]:.2f}
72
+ - Child: {gender_scores[2]:.2f}
73
  """
74
 
75
  return result