From 6fea7530dbe9d2ed4d81eea6d4cdee476e37a60d Mon Sep 17 00:00:00 2001 From: kadirnar Date: Thu, 25 Jan 2024 16:23:09 +0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20Fix=20return=20statement=20in=20?= =?UTF-8?q?ImageClassificationPipeline?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- computer_vision/transformers/image_classification/node.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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,)