2025-08-01 08:40 🚀

README.md

@@ -6,6 +6,7 @@ colorTo: pink
 sdk: gradio
 sdk_version: 5.39.0
 app_file: app.py
+suggested_hardware: t4-small
 pinned: true
 license: mit
 short_description: The crowd counting model ZIP-B

app.py

@@ -20,7 +20,8 @@ mean = (0.485, 0.456, 0.406)
 std = (0.229, 0.224, 0.225)
 alpha = 0.8
 EPS = 1e-8
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = torch.device("cuda")
 loaded_model = None
 current_model_config = {"variant": None, "dataset": None, "metric": None}
 
@@ -432,8 +433,9 @@ def predict(image: Image.Image, variant_dataset_metric: str):
     # den_map = normalize_map(den_map)
     # complete_zero_map = normalize_map(complete_zero_map)
     
-    # Apply a colormap (e.g., 'jet') to get an RGBA image
-    colormap = cm.get_cmap("jet")
+    # Apply a colormap for better visualization
+    # Options: 'viridis' (recommended), 'plasma', 'hot', 'inferno', 'jet'
+    colormap = cm.get_cmap("viridis")
 
     # The colormap returns values in [0,1]. Scale to [0,255] and convert to uint8.
     den_map = (colormap(den_map) * 255).astype(np.uint8)