dima806
/

diamond_types_image_detection

Image Classification

Inference Endpoints

Model card Files Files and versions Community

dima806 commited on Oct 19, 2024

Commit

073498a

·

verified ·

1 Parent(s): 22f2767

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -2,8 +2,10 @@
 license: apache-2.0
 metrics:
 - accuracy
 ---
-Returns diamond type given an image.
 See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.

 license: apache-2.0
 metrics:
 - accuracy
+base_model:
+- google/vit-base-patch16-224-in21k
 ---
+Returns diamond type given an image with about 99% accuracy.
 See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.