jazzmacedo commited on
Commit
177b95e
1 Parent(s): bef29fa

chore: update readme

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -7,7 +7,7 @@ tags:
7
  datasets:
8
  - imagefolder
9
  model-index:
10
- - name: vegetables-detector
11
  results:
12
  - task:
13
  name: Image Classification
@@ -27,7 +27,7 @@ language:
27
  pipeline_tag: image-classification
28
  ---
29
 
30
- # Vegetables Detector
31
 
32
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50).
33
 
@@ -57,7 +57,7 @@ from transformers import AutoModelForImageClassification
57
  from PIL import Image
58
 
59
  # Load the saved model and tokenizer
60
- model = AutoModelForImageClassification.from_pretrained("jazzmacedo/vegetables-detector")
61
 
62
  # Get the list of labels from the model's configuration
63
  labels = list(model.config.id2label.values())
 
7
  datasets:
8
  - imagefolder
9
  model-index:
10
+ - name: fruits-and-vegetables-detector-36
11
  results:
12
  - task:
13
  name: Image Classification
 
27
  pipeline_tag: image-classification
28
  ---
29
 
30
+ # fruits-and-vegetables-detector-36
31
 
32
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50).
33
 
 
57
  from PIL import Image
58
 
59
  # Load the saved model and tokenizer
60
+ model = AutoModelForImageClassification.from_pretrained("jazzmacedo/fruits-and-vegetables-detector-36")
61
 
62
  # Get the list of labels from the model's configuration
63
  labels = list(model.config.id2label.values())