Fix np.argmax arg

app.py

@@ -78,7 +78,7 @@ def generate_segmentation_video(study):
     # shape = (8, Z, H, W)
     p_spine = seg_output[:7].sum(0)
     # shape = (Z, H, W)
-    seg_output = np.argmax(seg_output, dim=0) + 1
+    seg_output = np.argmax(seg_output, axis=0) + 1
     # shape = (Z, H, W)
     seg_output[p_spine < 0.5] = 0
     seg_output = (seg_output * 255 / 7).astype("uint8")