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

  • Log In
  • Sign Up

arnaucas
/
wildfire-classifier

Image Classification
Transformers
PyTorch
Safetensors
vit
Generated from Trainer
climate
biology
Model card Files Files and versions
xet
Community

Instructions to use arnaucas/wildfire-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use arnaucas/wildfire-classifier with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-classification", model="arnaucas/wildfire-classifier")
    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("arnaucas/wildfire-classifier")
    model = AutoModelForImageClassification.from_pretrained("arnaucas/wildfire-classifier")
  • Notebooks
  • Google Colab
  • Kaggle
wildfire-classifier
689 MB
Ctrl+K
Ctrl+K
  • 2 contributors
History: 17 commits
arnaucas's picture
arnaucas
Upload pytorch_model.bin
57c12ca over 2 years ago
  • .gitattributes
    1.52 kB
    initial commit over 2 years ago
  • README.md
    1.83 kB
    Update README.md over 2 years ago
  • all_results.json
    348 Bytes
    wildfire_classifier over 2 years ago
  • config.json
    717 Bytes
    wildfire_classifier over 2 years ago
  • eval_results.json
    202 Bytes
    wildfire_classifier over 2 years ago
  • model.safetensors
    344 MB
    xet
    wildfire_classifier over 2 years ago
  • preprocessor_config.json
    327 Bytes
    wildfire_classifier over 2 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

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

    What is a pickle import?

    344 MB
    xet
    Upload pytorch_model.bin over 2 years ago
  • test_results.json
    200 Bytes
    wildfire_classifier over 2 years ago
  • train_results.json
    166 Bytes
    wildfire_classifier over 2 years ago
  • trainer_state.json
    5.17 kB
    wildfire_classifier over 2 years ago
  • training_args.bin

    Detected Pickle imports (8)

    • "transformers.trainer_utils.HubStrategy",
    • "transformers.training_args.OptimizerNames",
    • "accelerate.utils.dataclasses.DistributedType",
    • "accelerate.state.PartialState",
    • "torch.device",
    • "transformers.trainer_utils.SchedulerType",
    • "transformers.training_args.TrainingArguments",
    • "transformers.trainer_utils.IntervalStrategy"

    How to fix it?

    4.03 kB
    xet
    wildfire_classifier over 2 years ago