facebook
/

hiera-tiny-224-in1k-hf

Image Classification

Inference Endpoints

Model card Files Files and versions Community

EduardoPacheco commited on Jun 20, 2024

Commit

627ae3a

•

1 Parent(s): 302b97b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ import torch
 from PIL import Image
 from transformers import AutoImageProcessor, AutoModelForImageClassification
-model_id = "EduardoPacheco/hiera-tiny-224-in1k"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 image_processor = AutoImageProcessor.from_pretrained(model_id)

 from PIL import Image
 from transformers import AutoImageProcessor, AutoModelForImageClassification
+model_id = "facebook/hiera-tiny-224-in1k-hf"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 image_processor = AutoImageProcessor.from_pretrained(model_id)