jazzmacedo
commited on
Commit
•
177b95e
1
Parent(s):
bef29fa
chore: update readme
Browse files
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 |
-
#
|
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())
|