Update handler.py

handler.py

@@ -25,6 +25,8 @@ class EndpointHandler():
         
         # Concatenate all audio segments
         full_audio = torch.cat([torch.tensor(a) for a in audio_segments])
+
+        audio_length_seconds = len(full_audio) / sample_rate
         
         # Write full audio to a binary buffer
         buffer = io.BytesIO()
@@ -39,5 +41,6 @@ class EndpointHandler():
             },
             "body": audio_b64,
             "statusCode": 200,
-            "isBase64Encoded": True  # if binary, encode
+            "isBase64Encoded": True,
+            "audio_length_seconds": audio_length_seconds
         }