diff --git a/main.py b/main.py
index 32bf3c1..a3fb6d0 100644
--- a/main.py
+++ b/main.py
@@ -57,7 +57,7 @@ async def endpoint(body : Predictbody):
     result = None
 
     try:
-        result = engine.inference(res)
+        result = engine.inference_batch_json_data(res)
     except Exception as e:
         result = None