NeuronZero
/

WBC-Classifier

Image Classification

Model card Files Files and versions

NeuronZero commited on Apr 4, 2024

Commit

1da060a

·

verified ·

1 Parent(s): 43db832

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -57,4 +57,5 @@ inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)
 logits = outputs.logits
 predicted_class_idx = logits.argmax(-1).item()
-print("Predicted class:", model.config.id2label[predicted_class_idx])

 outputs = model(**inputs)
 logits = outputs.logits
 predicted_class_idx = logits.argmax(-1).item()
+print("Predicted class:", model.config.id2label[predicted_class_idx])
+```