🚀 Fix return statement in

ImageClassificationPipeline
2024-01-25 16:23:09 +03:00
parent caa1dfec72
commit 6fea7530db
1 changed files with 1 additions and 2 deletions
@@ -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,)