Update server1.py

server1.py

@@ -38,8 +38,14 @@ def detect_blocks(image_bytes):
 
     for mask in masks:
 
-        # ***** ARREGLO AQUI *****
-        mask = np.squeeze(mask)  # <-- MUY IMPORTANTE
+        mask = np.array(mask)
+
+        # forzar a 2D
+        while mask.ndim > 2:
+            mask = mask[0]
+
+        if mask.ndim != 2:
+            continue
 
         ys, xs = np.where(mask > 0.5)
         if len(xs) == 0:
@@ -53,7 +59,6 @@ def detect_blocks(image_bytes):
     return blocks
 
 
-
 @app.post("/strip/")
 async def strip(front: UploadFile = File(...)):
     try: