Update README.md

README.md

@@ -46,10 +46,10 @@ outputs = model(**inputs)
 predicted_class_idx = np.argmax(outputs.logits[0]).item()
 
 # Define class labels
-id2label = {0: "non-vascular", 1: "vascular"}
+label = {0: "non-vascular", 1: "vascular"}
 
 # Get predicted class label
-predicted_class_label = id2label[predicted_class_idx]
+predicted_class_label = label[predicted_class_idx]
 print(predicted_class_label)
 ```
 
@@ -72,10 +72,10 @@ outputs = model(**inputs)
 predicted_class_idx = np.argmax(outputs.logits[0]).item()
 
 # Define class labels
-id2label = {0: "non-vascular", 1: "vascular"}
+label = {0: "non-vascular", 1: "vascular"}
 
 # Get predicted class label
-predicted_class_label = id2label[predicted_class_idx]
+predicted_class_label = label[predicted_class_idx]
 print(predicted_class_label)
 ```