Update README.md

README.md

@@ -59,6 +59,26 @@ More information on the base model used can be found here: (https://huggingface.
 
 ## How to use this Model
 
+```python
+!pip install transformers --quiet
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from PIL import Image
+import requests
+
+url = 'https://sdo.gsfc.nasa.gov/assets/gallery/preview/211_coronalhole.jpg'
+image = Image.open(requests.get(url, stream=True).raw)
+
+feature_extractor = AutoFeatureExtractor.from_pretrained("kenobi/SDO_VT1")
+model = AutoModelForImageClassification.from_pretrained("kenobi/SDO_VT1")
+inputs = feature_extractor(images=image, return_tensors="pt")
+
+outputs = model(**inputs)
+logits = outputs.logits
+# model predicts one of the three fine-tuned classes (NASA_SDO_Coronal_Hole, NASA_SDO_Coronal_Loop or NASA_SDO_Solar_Flare) 
+predicted_class_idx = logits.argmax(-1).item()
+print("Predicted class:", model.config.id2label[predicted_class_idx])
+```
+
 ## References
 
 A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation: