small bug fix

app.py

@@ -66,7 +66,7 @@ transform = transforms.Compose(
 def get_overlay(image: torch.Tensor, preds: torch.Tensor, alpha: float) -> torch.Tensor:
     """Generates the segmentation ovelay for an satellite image"""
 
-    masks = preds.squeeze().label_to_onehot(masks, 7)
+    masks = label_to_onehot(preds.squeeze(), 7)
     overlay = draw_segmentation_masks(
         image, masks=masks, alpha=alpha, colors=class_rgb_colors
     )