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 |
|