diff --git a/computer_vision/transformers/image_classification/node.py b/computer_vision/transformers/image_classification/node.py index fbb2584..f7b99f8 100644 --- a/computer_vision/transformers/image_classification/node.py +++ b/computer_vision/transformers/image_classification/node.py @@ -49,5 +49,4 @@ class ImageClassificationPipeline: predicted_label_name = model.config.id2label[predicted_label_id] predicted_score = torch.softmax(logits, -1)[0, predicted_label_id].item() - return predicted_label_name, predicted_score - + return (predicted_label_name, predicted_score,)