Update handler.py

handler.py

@@ -40,6 +40,8 @@ class EndpointHandler:
         logger.debug(f"Audio: {audio}")
         sampling_rate = parameters.pop("sampling_rate", None)
         logger.debug(f"Sampling Rate: {sampling_rate}")
+        if not sampling_rate:
+            sampling_rate = self.model.config.audio_encoder.sampling_rate
 
         if audio is not None:
             audio_list = json.loads(audio)
@@ -77,4 +79,4 @@ class EndpointHandler:
         # postprocess the prediction
         prediction = outputs[0].cpu().numpy()
 
-        return [{"generated_text": prediction}]
+        return [{"generated_text": prediction, "sampling_rate" : sampling_rate}]