Update README.md

README.md

@@ -2,4 +2,28 @@
 license: apache-2.0
 base_model:
 - facebook/dinov2-with-registers-small
----
+---
+
+### Example
+# Load model directly
+```python
+from transformers import AutoImageProcessor, Dinov2WithRegistersForImageClassification
+import torch
+import numpy as np
+from PIL import Image
+import os
+import pandas as pd
+
+image_processor = AutoImageProcessor.from_pretrained('WpythonW/dinoV2-deepfake-detector')
+model = Dinov2WithRegistersForImageClassification.from_pretrained("WpythonW/dinoV2-deepfake-detector")
+model.config.id2label = {0: "FAKE", 1: "REAL"}
+model.config.label2id = {"FAKE": 0, "REAL": 1}
+
+real = Image.open('real.jpg')
+fake = Image.open('fake1.png')
+
+inputs = image_processor([real, fake], return_tensors="pt")
+
+with torch.no_grad():
+    logits = model(**inputs).logits
+```