Update README.md

README.md

@@ -59,7 +59,33 @@ This checkpoint of EfficientFormer-L1 was trained for 1000 epochs.
 Use the code below to get started with the model.
 
 ```python
-# A nice code snippet here that describes how to use the model...
+import requests
+import torch
+from PIL import Image
+
+from transformers import EfficientFormerImageProcessor, EfficientFormerForImageClassificationWithTeacher
+
+# Load a COCO image of two cats to test the model
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+
+# Load preprocessor and pretrained model
+model_name = "huggingface/efficientformer-l1-300"
+processor = EfficientFormerImageProcessor.from_pretrained(model_name)
+model = EfficientFormerForImageClassificationWithTeacher.from_pretrained(model_name)
+
+# Preprocess input image
+inputs = processor(images=image, return_tensors="pt")
+
+# Inference
+with torch.no_grad():
+	outputs = model(**inputs)
+
+# Print the top ImageNet1k class prediction 
+logits = outputs.logits
+scores = torch.nn.functional.softmax(logits, dim=1)
+top_pred_class = torch.argmax(scores, dim=1)
+print(f"Predicted class: {top_pred_class}")
 ```
 </how_to_start>