Fix bug with values

app.py

@@ -31,13 +31,16 @@ def compute_depth(image, preds):
         align_corners=False,
     ).squeeze().numpy()
 
+    print(prediction.shape)
+
     output = []
 
     for pred in preds:
         x = (pred["box"]["xmax"] - pred["box"]["xmin"]) // 2
         y = (pred["box"]["ymax"] - pred["box"]["ymin"]) // 2
         output.append({
-            "distance": float(prediction[x][y])
+            "class": pred["label"],
+            "distance": float(prediction[y][x])
         })
 
     return output