Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

andreotte
/
vit-multi-label

Image Classification
Transformers
PyTorch
vit
Model card Files Files and versions
xet
Community
1

Instructions to use andreotte/vit-multi-label with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use andreotte/vit-multi-label with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-classification", model="andreotte/vit-multi-label")
    pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
    # Load model directly
    from transformers import AutoImageProcessor, AutoModelForImageClassification
    
    processor = AutoImageProcessor.from_pretrained("andreotte/vit-multi-label")
    model = AutoModelForImageClassification.from_pretrained("andreotte/vit-multi-label")
  • Notebooks
  • Google Colab
  • Kaggle
vit-multi-label
Ctrl+K
Ctrl+K
  • 1 contributor
History: 3 commits
andreotte's picture
andreotte
Upload feature extractor
059cec1 over 3 years ago
  • .gitattributes
    1.48 kB
    initial commit over 3 years ago
  • config.json
    880 Bytes
    Upload ViTForImageClassification over 3 years ago
  • preprocessor_config.json
    228 Bytes
    Upload feature extractor over 3 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

    • "torch._utils._rebuild_tensor_v2",
    • "torch.FloatStorage",
    • "collections.OrderedDict"

    What is a pickle import?

    343 MB
    xet
    Upload ViTForImageClassification over 3 years ago