Fix handler

handler.py

@@ -3,19 +3,26 @@ import torch
 import numpy as np
 import scipy
 import io
+import base64
 
-preload_models()
+class EndpointHandler:
+    def __init__(self, path=""):
+        preload_models()
 
-def infer(inputs):
-    text = inputs.get("text", "")
-    if not text.strip():
-        return {"error": "No input text provided."}
+    def __call__(self, data):
+        text = data.get("inputs", "")
+        if not text.strip():
+            return {"error": "No input text provided."}
 
-    audio_array = generate_audio(text, history_prompt="v2/ru_speaker_0")
+        audio_array = generate_audio(text, history_prompt="v2/ru_speaker_0")
 
-    # Сохраняем в wav-поток
-    byte_io = io.BytesIO()
-    scipy.io.wavfile.write(byte_io, rate=24000, data=audio_array)
-    byte_io.seek(0)
+        byte_io = io.BytesIO()
+        scipy.io.wavfile.write(byte_io, rate=24000, data=audio_array)
+        byte_io.seek(0)
 
-    return {"audio": byte_io.read()}
+        audio_base64 = base64.b64encode(byte_io.read()).decode("utf-8")
+
+        return {
+            "audio_base64": audio_base64,
+            "content_type": "audio/wav"
+        }