Upload model

modeling.py

@@ -391,6 +391,6 @@ class CTCropModel(PreTrainedModel):
         if remove_empty_slices and empty.sum() > 0:
             empty_indices = list(torch.where(empty)[0].cpu().numpy())
             print(f"removing {empty.sum()} empty slices ...")
-            cropped = cropped[~empty]
+            cropped = cropped[~empty.cpu().numpy()]
             return cropped, empty_indices
         return cropped