Update app.py

app.py

@@ -48,10 +48,9 @@ def process_frame(image):
     predicted_depth = model(input_tensor).predicted_depth
     depth_map = predicted_depth.squeeze()
     
-    # Discretization on GPU
-    num_bins = 256  # Match the color_map size
-    depth_map = torch.quantile(depth_map, torch.linspace(0, 1, num_bins, device=device))
-    depth_map = torch.bucketize(depth_map.squeeze(), depth_map)
+    # Normalize depth map to [0, 255] range
+    depth_map = (depth_map - depth_map.min()) / (depth_map.max() - depth_map.min())
+    depth_map = (depth_map * 255).byte()
     
     depth_map_colored = color_map[depth_map]