Update server.py

server.py

@@ -55,7 +55,7 @@ print(f"✓ Image transform defined (size: {CONFIG['image_size']}x{CONFIG['image
 def preprocess_image(image_path: str) -> torch.Tensor:
     """Load and preprocess image."""
     image = Image.open(image_path).convert('RGB')
-    return transform(image)
+    return transform(image).unsqueeze(0)  # Add batch dimension
 
 # ─────────────────────────────────────────────────────────────────────────────
 # ARCHITECTURE 1 — CoAtNet Encoder  (shared by all three models)