Upload folder using huggingface_hub

app.py

@@ -27,6 +27,7 @@ import subprocess
 import sys
 import tempfile
 
+import matplotlib
 import numpy as np
 import spaces
 import torch
@@ -56,6 +57,10 @@ from omegaconf import OmegaConf
 from src.utils.geometry_utils import erp_to_cubemap
 from src.utils.utils import prepare_depth_for_logging
 
+# `prepare_depth_for_logging` *calls* its cmap argument, so it needs a
+# colormap object; a name string raises "'str' object is not callable".
+CMAP = matplotlib.colormaps["Spectral"]
+
 CHECKPOINT = "prs-eth/PaGeR-metric-depth"
 MAX_WIDTH = 2048          # our panoramas are 2K native; PaGeR accepts up to 3K
 
@@ -106,7 +111,7 @@ def predict(image_path: str):
     depth, preview = None, None
     try:
         out = prepare_depth_for_logging(
-            pager, pred["depth"][0], None, (height, width), "Spectral"
+            pager, pred["depth"][0], None, (height, width), CMAP
         )
         depth, preview = (out if isinstance(out, (tuple, list)) else (out, None))[:2]
     except Exception as exc:                                  # pragma: no cover
@@ -118,8 +123,12 @@ def predict(image_path: str):
         depth = depth.detach().float().cpu().numpy()
     depth = np.squeeze(np.asarray(depth, dtype=np.float32))
     if depth.shape != (height, width):
+        # pred["depth"] is the per-face cubemap (6, face, face); the merge back
+        # to equirect happens inside prepare_depth_for_logging, so this means
+        # that call failed and the fallback cannot substitute for it.
         raise gr.Error(
-            f"depth came back as {depth.shape}, expected {(height, width)}"
+            f"depth came back as {depth.shape}, expected {(height, width)} — "
+            "the equirect merge did not run; check the Space logs."
         )
 
     if preview is None:

requirements.txt

@@ -11,3 +11,4 @@ huggingface_hub
 omegaconf
 numpy
 pillow
+matplotlib