Hugging Face's logo

NeuronZero
/
WBC-Classifier

Image Classification
Transformers
Safetensors
resnet
auto-train
vision
Model card Files
xet
 Community
NeuronZero commited on
Commit
1da060a
·
verified ·
1 Parent(s): 43db832

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -57,4 +57,5 @@ inputs = processor(images=image, return_tensors="pt")
57
  outputs = model(**inputs)
58
  logits = outputs.logits
59
  predicted_class_idx = logits.argmax(-1).item()
60
- print("Predicted class:", model.config.id2label[predicted_class_idx])
 
 
57
  outputs = model(**inputs)
58
  logits = outputs.logits
59
  predicted_class_idx = logits.argmax(-1).item()
60
+ print("Predicted class:", model.config.id2label[predicted_class_idx])
61
+ ```