Update tasks/audio.py

tasks/audio.py

@@ -44,6 +44,10 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     # Split dataset
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
+    def compute_spectrogram(audio_array, sample_rate=16000, frame_length=256, frame_step=128):
+        spectrogram = tf.signal.stft(audio_array, frame_length=frame_length, frame_step=frame_step)
+        spectrogram = tf.abs(spectrogram)
+        return tf.expand_dims(spectrogram, axis=-1)
     def preprocess_dataset(dataset, max_length=16000):
         def preprocess_item(item):
             audio_array = item["audio"]["array"]