| --- |
| license: apache-2.0 |
| --- |
| |
| Use at your own risk: |
|
|
| ```python |
| from transformers import AutoFeatureExtractor, AutoModelForImageClassification |
| from datasets import load_dataset |
| import torch |
| |
| feature_extractor = AutoFeatureExtractor.from_pretrained("fxmarty/tiny-testing-remote-code") |
| |
| model = AutoModelForImageClassification.from_pretrained("fxmarty/tiny-testing-remote-code", trust_remote_code=True) |
| |
| dataset = load_dataset("huggingface/cats-image") |
| image = dataset["test"]["image"][0] |
| |
| inputs = feature_extractor(image, return_tensors="pt") |
| |
| with torch.no_grad(): |
| logits = model(**inputs).logits |
| |
| # model predicts one of the 1000 ImageNet classes |
| predicted_label = logits.argmax(-1).item() |
| print(model.config.id2label[predicted_label]) |
| ``` |
|
|
