metadata

tags:
  - autotrain
  - image-classification
  - pytorch
  - transformers
library_name: pytorch
base_model: microsoft/resnet-50
widget:
  - src: >-
      https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
    example_title: Tiger
  - src: >-
      https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
    example_title: Teapot
  - src: >-
      https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
    example_title: Palace
datasets:
  - A2H0H0R1/plant-disease-new
license: apache-2.0

#Inference Pipeline

from transformers import AutoModelForImageClassification, AutoProcessor

model = AutoModelForImageClassification.from_pretrained("ozair23/autotrain-w5nk2-rvmqx") processor = AutoProcessor.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")

def predict(image): inputs = processor(images=image, return_tensors="pt") outputs = model(**inputs) return outputs

Model Trained Using AutoTrain

Problem type: Image Classification

Validation Metrics

No validation metrics available