Update handler.py

handler.py

@@ -15,33 +15,23 @@ class EndpointHandler():
         inputs = data.get("inputs", {})
         text = inputs.get("text")
         voice = inputs.get("voice")
-
+        
         audio_segments = []
-
         generator = self.pipeline(text, voice)
+        
+        # Direct append without saving to disk
         for i, (gs, ps, audio) in enumerate(generator):
-            # Save temporarily just in case (optional if you skip saving to disk)
-            filename = f"{i}.wav"
-            sf.write(filename, audio, 24000)
             audio_segments.append(audio)
-
+        
         # Concatenate all audio segments
         full_audio = torch.cat([torch.tensor(a) for a in audio_segments])
-
+        
         # Write full audio to a binary buffer
         buffer = io.BytesIO()
         sf.write(buffer, full_audio.numpy(), 24000, format='WAV')
         buffer.seek(0)
-
-        # Clean up temp files
-        for i in range(len(audio_segments)):
-            try:
-                os.remove(f"{i}.wav")
-            except FileNotFoundError:
-                pass
-
+        
         from flask import Response
-        buffer.seek(0)
         return Response(
             buffer.read(),
             mimetype='audio/wav',