 model predicts bounding boxes and corresponding face mask detection classes logits = outputs.logits bboxes = outputs.pred_boxes ```  Currently, both the feature extractor and model support PyTorch.   