Add check for audio sampling rate other than 8k or 16k Hz (#898)

* add check for sr other than 8k or 16k * update error message to mention resampling * return error status --------- Co-authored-by: Sayan Shaw <[email protected]>
microsoft · Feb 28, 2025 · 5541566 · 5541566
1 parent f63857d
commit 5541566
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/shared/api/speech_features.hpp b/shared/api/speech_features.hpp
@@ -623,7 +623,13 @@ class Phi4AudioEmbed {
 
     SpeechLibLogMel logmel;
     // already checked in Init
-    logmel.Init(sr_val == 8000? logmel_8k_attrs_: logmel_attrs_);
+
+    // Currently we only support 8k and 16k Hz sampling rate.
+    if (sr_val != 8000 && sr_val != 16000){
+      return OrtxStatus(kOrtxErrorNotImplemented, "Currently only 8k and 16k Hz sampling rate is supported. Please resample your audio file with unsupported audio sampling rate: " + sr_val);
+    }
+
+    logmel.Init(sr_val == 8000 ? logmel_8k_attrs_: logmel_attrs_);
     status = logmel.Compute(stft_norm, ts_logmel);
     if (!status.IsOk()) {
       return status;