Update README.md

README.md

@@ -40,7 +40,36 @@ To load and use the model in your Python environment, use the standard `transfor
 
 ```bash
 pip install transformers torch pillow
+```
 
+```python
+from transformers import AutoModelForImageClassification, AutoFeatureExtractor
+from PIL import Image
+import torch
+
+image_path = 'path/to/your/image.jpg' 
+model_id = "SADRACODING/SDXL-Deepfake-Detector"
+
+# Load Model and Feature Extractor
+model = AutoModelForImageClassification.from_pretrained(model_id)
+feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)
+
+# Preprocessing
+image = Image.open(image_path).convert("RGB")
+inputs = feature_extractor(images=image, return_tensors="pt")
+
+# Inference
+with torch.no_grad():
+    logits = model(**inputs).logits
+
+# Post-processing and Prediction
+predicted_class_id = logits.argmax().item()
+labels = ["REAL", "FAKE"] 
+prediction = labels[predicted_class_id]
+confidence = torch.nn.functional.softmax(logits, dim=-1)[0][predicted_class_id].item()
+
+print(f"Prediction: {prediction} (Confidence: {confidence:.4f})")
+```
 ---
 license: mit
 ---