only display middle slice for 3D images

app.py

@@ -155,6 +155,11 @@ def apply_windowing(image: np.ndarray, subset: str) -> np.ndarray:
 def to_display_image(image: np.ndarray) -> np.ndarray:
     """Normalise image for display purposes (uint8, 3-channel)."""
 
+    # if image is 3D, keep the middle slice
+    if image.ndim == 3:
+        gr.Info(f"Image is 3D, we display only the middle slice")
+        image = image[:, :, image.shape[2] // 2]
+
     arr = np.array(image, copy=True)
     if not np.isfinite(arr).all():
         arr = np.nan_to_num(arr, nan=0.0)