dima806 commited on
Commit
d31a228
1 Parent(s): d648271

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
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