Update README.md
Browse files
README.md
CHANGED
|
@@ -10,6 +10,8 @@ language:
|
|
| 10 |
metrics:
|
| 11 |
- accuracy
|
| 12 |
- f1
|
|
|
|
|
|
|
| 13 |
---
|
| 14 |
|
| 15 |
# Umsakwa/Uddayvit-image-classification-model
|
|
@@ -85,4 +87,4 @@ image = processor(images="path_to_image.jpg", return_tensors="pt")
|
|
| 85 |
|
| 86 |
# Run inference
|
| 87 |
outputs = model(**image)
|
| 88 |
-
predictions = outputs.logits.argmax(-1)
|
|
|
|
| 10 |
metrics:
|
| 11 |
- accuracy
|
| 12 |
- f1
|
| 13 |
+
datasets:
|
| 14 |
+
- AI-Lab-Makerere/beans
|
| 15 |
---
|
| 16 |
|
| 17 |
# Umsakwa/Uddayvit-image-classification-model
|
|
|
|
| 87 |
|
| 88 |
# Run inference
|
| 89 |
outputs = model(**image)
|
| 90 |
+
predictions = outputs.logits.argmax(-1)
|