Update README.md

README.md

@@ -10,6 +10,7 @@ metrics:
 
 # Vision Transformer fine-tuned on kvasir_v2 for colonoscopy classification
 
+## Demo
 
 ### Drag the following images to the widget to test the model
 
@@ -22,6 +23,11 @@ metrics:
 - ![](https://i.imgur.com/qcCYpN9_d.webp?maxwidth=224&fidelity=grand)
 
 
+## Training
+
+You can find the code [here](https://github.com/qanastek/HugsVision/blob/main/recipes/kvasir_v2/binary_classification/Kvasir_v2_Image_Classifier.ipynb)
+
+
 ## Metrics
 
 ```