Update server.py

server.py

@@ -27,24 +27,29 @@ async def root():
     with open("index.html", "r", encoding="utf-8") as f:
         return HTMLResponse(f.read())
 
-
 @app.post("/infer")
 async def run_inference(images: list[UploadFile] = File(...)):
-    # save uploads
     img_paths = []
-    for img in images:
-        tmp = f"/tmp/{img.filename}"
-        with open(tmp, "wb") as f:
-            f.write(await img.read())
+    for upload in images:
+        data = await upload.read()
+
+        # Decode bytes → BGR image, dropping any alpha channel
+        bgr = mmcv.imfrombytes(data, flag="color")  
+
+        # Overwrite to disk as a 3-channel PNG (or JPEG)
+        tmp = f"/tmp/{upload.filename}"
+        mmcv.imwrite(bgr, tmp)
+
         img_paths.append(tmp)
 
-    # run your BEV inference
+    # Now all inputs are (3, H, W)—your model won’t hit that dim==4 assertion
     bev_paths = infer_images(img_paths)
 
-    # base64-encode results
+    # Base64-encode outputs
     output = []
     for p in bev_paths:
         with open(p, "rb") as f:
             b64 = base64.b64encode(f.read()).decode("utf-8")
         output.append({"bev_image": b64})
+
     return JSONResponse(content=output)