Update handler.py

handler.py

@@ -1,5 +1,6 @@
 from typing import Dict
 from faster_whisper import WhisperModel
+import io
 
 class EndpointHandler:
     def __init__(self, model_dir=None):
@@ -11,8 +12,11 @@ class EndpointHandler:
         # Process the input data expected to be in 'inputs' key containing audio file bytes
         audio_bytes = data["inputs"]
         
+        # Convert bytes to a file-like object
+        audio_file = io.BytesIO(audio_bytes)
+        
         # Perform transcription using the model
-        segments, info = self.model.transcribe(audio_bytes)
+        segments, info = self.model.transcribe(audio_file)
         
         # Compile the results into a text string and extract language information
         text = " ".join(segment.text for segment in segments)