how to use the model

README.md

@@ -73,4 +73,28 @@ accuracy: 0.737093690248566 -->
 |------|-------|
 | Macro | 0.4972 |
 | Micro | 0.7371 |
-| Weighted | 0.7371 |
+| Weighted | 0.7371 |
+
+
+## How to use
+
+This model is designed for image classification. Here's how you can use it:
+
+```python
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+import torch
+from PIL import Image
+
+model_name = "eligapris/v-mdd-2000"
+processor = AutoImageProcessor.from_pretrained(model_name)
+model = AutoModelForImageClassification.from_pretrained(model_name)
+
+image = Image.open("path_to_your_image.jpg")
+inputs = processor(images=image, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model(**inputs)
+
+logits = outputs.logits
+predicted_class_idx = logits.argmax(-1).item()
+print("Predicted class:", model.config.id2label[predicted_class_idx])