Update README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,8 @@ license: apache-2.0
|
|
| 3 |
metrics:
|
| 4 |
- accuracy
|
| 5 |
- f1
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |
Returns mushroom type given an image.
|
| 8 |
|
|
|
|
| 3 |
metrics:
|
| 4 |
- accuracy
|
| 5 |
- f1
|
| 6 |
+
base_model:
|
| 7 |
+
- google/vit-base-patch16-224-in21k
|
| 8 |
---
|
| 9 |
Returns mushroom type given an image.
|
| 10 |
|