Added Model Details

README.md

@@ -18,35 +18,39 @@ model-index:
         value: 0.9107142686843872
 ---
 
-# big-cat-classifier
+# Big Cat Classifier 🐯
 
+An image classifier that categories between the 5 members of the big cat family i.e.
 
-Autogenerated by HuggingPics🤗🖼️
+1) __Cheetah__
 
-Create your own image classifier for **anything** by running [the demo on Google Colab](https://colab.research.google.com/github/nateraw/huggingpics/blob/main/HuggingPics.ipynb).
+![cheetah](images/cheetah.jpg)
 
-Report any issues with the demo at the [github repo](https://github.com/nateraw/huggingpics).
+2) __Jaguar__
 
+![jaguar](images/jaguar.jpg)
 
-## Example Images
+3) __Leopard__
 
+![leopard](images/leopard.jpg)
 
-#### cheetah
+4) __Lion__
 
-![cheetah](images/cheetah.jpg)
+![lion](images/lion.jpg)
 
-#### jaguar
+5) __Tiger__
 
-![jaguar](images/jaguar.jpg)
+![tiger](images/tiger.jpg)
 
-#### leopard
 
-![leopard](images/leopard.jpg)
+A visual transformer is used as the base model. After fine-tuning, the classifier was trained for 10 epochs and obtained an accuracy of 0.911
 
-#### lion
+> __Note__:
+> - While the identification of a lion or a tiger is fairly simple, the ability of the model is tested when it has to accurately distinguish between a leopard, cheetah, and jaguar
+> - Since the model has been trained on a small dataset of approximately 500 images (in total), testing on various subspecies found among the populations of each member of the big cats is to be done. For example, The robustness of the model to classify all Bengal Tigers, Siberian Tigers, Indochinese Tigers, and Malayan Tigers as Tigers
+> - Lastly, the performance of the model in categorizing certain rare variants in the populations of big cats such as white tigers, snow leopards or black jaguar has not been determined exclusively.
 
-![lion](images/lion.jpg)
 
-#### tiger
+# Reference and Acknowledgement:
 
-![tiger](images/tiger.jpg)
+[Hugging Pics](https://github.com/nateraw/huggingpics)