Update README.md

README.md

@@ -1,3 +1,13 @@
+---
+license: mit
+language:
+- en
+base_model:
+- sakshamkr1/ResNet50-APTOS-DR
+pipeline_tag: image-classification
+tags:
+- onnx
+---
 # ResNet50-APTOS-DR (ONNX) - Single Clean File
 
 **5-class Diabetic Retinopathy classifier** ready for Raspberry Pi 5 and edge devices.
@@ -36,7 +46,7 @@ transform = transforms.Compose([
     transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
 ])
 
-img = Image.open("DR1.jpg").convert("RGB")
+img = Image.open("Your_image").convert("RGB")
 input_tensor = transform(img).unsqueeze(0).numpy().astype(np.float32)
 
 output = session.run(None, {"input": input_tensor})[0][0]
@@ -48,5 +58,4 @@ print(f"✅ Predicted: {classes[pred_idx]} ({probs[pred_idx]*100:.1f}%)")
 ```
 
 **License**: MIT  
-Made for low-resource diabetic retinopathy screening ❤️
-
+Made for low-resource diabetic retinopathy screening ❤️