Update handler.py

handler.py

@@ -24,9 +24,7 @@ class EndpointHandler:
 
             AUDIO_PROMPT_PATH=hf_hub_download(repo_id="aiplexdeveloper/chatterbox", filename="arjun_das_output_audio.mp3")
             wav = self.model.generate(text, audio_prompt_path=AUDIO_PROMPT_PATH, exaggeration = exaggeration, cfg_weight=cfg_weight)
-            
-            # ta.save("test-2.wav", wav, self.model.sr)
-            # Convert the tensor to numpy and write to an in-memory buffer
+
             buffer = io.BytesIO()
             sf.write(buffer, wav.cpu().numpy().T, self.model.sr, format='WAV')
             buffer.seek(0)
@@ -34,7 +32,10 @@ class EndpointHandler:
             # Encode to base64
             audio_base64 = base64.b64encode(buffer.read()).decode('utf-8')
 
-            return [{"audio_base64": audio_base64}]
+            audio_length_seconds = len(wav) / self.model.sr
+
+
+            return [{"audio_base64": audio_base64, "audio_length_seconds":audio_length_seconds}]
 
 
         except Exception as e: