updated README.

added Dataset section.

README.md

@@ -73,3 +73,7 @@ output = model(input_data)["output_0"]
 # Prints the shape of the output tensor for verification
 print("Output shape:", output.shape)
 ```
+
+## Dataset
+
+The model was trained on the LGG Segmentation Dataset, which contains Brain MRI images with FLAIR abnormalities labeled for segmentation tasks. The dataset is available on [Kaggle](https://www.kaggle.com/datasets/mateuszbuda/lgg-mri-segmentation).