Update README.md

README.md

@@ -20,7 +20,7 @@ This repository contains a fine-tuned **Vision Transformer (ViT)** model for cla
 
 ## Model Overview
 
-The fine-tuned model is based on the **Vision Transformer (ViT)** architecture, which excels in handling image-based tasks by leveraging attention mechanisms for efficient feature extraction. The model was trained on the **CheXpert dataset**, which consists of labeled chest X-ray images for detecting diseases such as pneumonia, cardiomegaly, and others.
+The fine-tuned [Model Code on github](https://github.com/TheSmartTechnologyLab/vit-chest-xray) is based on the **Vision Transformer (ViT)** architecture, which excels in handling image-based tasks by leveraging attention mechanisms for efficient feature extraction. The model was trained on the **CheXpert dataset**, which consists of labeled chest X-ray images for detecting diseases such as pneumonia, cardiomegaly, and others.
 
 ## Performance