Update README.md

README.md

@@ -11,9 +11,10 @@ tags:
 - medical-imaging
 base_model:
 - microsoft/resnet-50
+pipeline_tag: image-classification
 ---
 
-# Diabetic Retinopathy Detection Model
+# Diabetic Retinopathy Detection Model ![LICENSE](https://img.shields.io/badge/CC--BY--SA--4.0-lightgrey?style=for-the-badge) ![PyTorch](https://img.shields.io/badge/PyTorch-EE4C2C?style=for-the-badge&logo=pytorch&logoColor=white)
 
 ## Overview
 
@@ -82,6 +83,4 @@ print(f"Predicted class: {predicted_class}")
 ```
 
 ## License
-This model is released under the **CC-BY-NC 4.0** license.
-
-
+This model is released under the **CC-BY-NC 4.0** license.