diff --git a/src/htrflow_core/models/huggingface/dit.py b/src/htrflow_core/models/huggingface/dit.py index 61158e2..0294689 100644 --- a/src/htrflow_core/models/huggingface/dit.py +++ b/src/htrflow_core/models/huggingface/dit.py @@ -78,6 +78,4 @@ def _label_based_on_return_format(self, logits): else: probabilities = torch.nn.functional.softmax(logits, dim=-1) label_ = {self.model.config.id2label[id]: prob.item() for id, prob in enumerate(probabilities)} - - logger.info(f"Prediction complete. Return format: {self.return_format}, label: {label_}.") return label_