Update README.md

README.md

@@ -71,7 +71,7 @@ inputs = processor(image, return_tensors="pt")
 with torch.no_grad():
     logits = model(**inputs).logits
 
-# model predicts one of the 1000 ImageNet classes
+# model predicts either glaucoma or non-glaucoma.
 predicted_label = logits.argmax(-1).item()
 print(model.config.id2label[predicted_label])