Update handler.py

handler.py

@@ -28,6 +28,7 @@ class EndpointHandler:
         parameters = data.pop("parameters", None)
         duration = parameters.pop("duration", None)
         audio = parameters.pop("audio", None)
+        sampling_rate = parameters.pop("sampling_rate", None)
 
         if audio is not None:
             audio_list = json.loads(audio)
@@ -45,13 +46,16 @@ class EndpointHandler:
             text=[inputs],
             padding=True,
             return_tensors="pt",
-            audio=audio).to(self.device)
+            audio=audio,
+            sampling_rate=sampling_rate).to(self.device)
 
         # If 'duration' is inside 'parameters', remove it
         if parameters is not None and 'duration' in parameters:
             parameters.pop('duration')
         if parameters is not None and 'audio' in parameters:
             parameters.pop('audio')
+        if parameters is not None and 'sampling_rate' in parameters:
+            parameters.pop('sampling_rate')
 
         # pass inputs with all kwargs in data
         if parameters is not None: