Update inference.py

inference.py

@@ -1,16 +1,34 @@
 from PIL import Image
 import numpy as np
+import base64
+from io import BytesIO
 
-class Pipeline:  # <--- обязательно Pipeline
+class Pipeline:
     def __init__(self, path=""):
-        print("Model initialized")
+        print("Pipeline initialized")
 
     def __call__(self, inputs):
-        image = inputs.get("image")
+        image_data = inputs.get("image")
 
-        if isinstance(image, str):
-            from base64 import b64decode
-            from io import BytesIO
-            image = Image.open(BytesIO(b64decode(image)))
+        if isinstance(image_data, str):
+            image_bytes = base64.b64decode(image_data)
+            image = Image.open(BytesIO(image_bytes))
+        else:
+            return {"error": "No image provided"}
 
-        return {"msg": "Got image", "size": image.size}
+        width, height = image.size
+
+        # Пример простой маски: прямоугольник по центру
+        mask = np.zeros((height, width), dtype=np.uint8)
+        mask[height//4: 3*height//4, width//4: 3*width//4] = 1
+
+        # Преобразуем в обычный Python-список
+        mask_list = mask.tolist()
+
+        return [
+            {
+                "label": "mock-segment",
+                "mask": mask_list,
+                "score": 0.99
+            }
+        ]