Update README.md

README.md

@@ -37,7 +37,22 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+This model was created by importing the dataset of the photos of ECG image into Google Colab from kaggle here: https://www.kaggle.com/datasets/erhmrai/ecg-image-data/data . I then used the image classification tutorial here: https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/image_classification.ipynb 
+
+obtaining the following notebook:
+
+https://colab.research.google.com/drive/1KC6twirtsc7N1kmlwY3IQKVUmSuK7zlh?usp=sharing
+
+The possible classified data are:
+<ul>
+<li>N: Normal beat</li>
+<li>S: Supraventricular premature beat</li>
+<li>V: Premature ventricular contraction</li>
+<li>F: Fusion of ventricular and normal beat</li>
+<li>Q: Unclassifiable beat</li>
+<li>M: myocardial infarction</li>
+</ul>
+
 
 ## Intended uses & limitations