WpythonW's picture
Update README.md
19de24b verified
---
license: apache-2.0
base_model:
- facebook/dinov2-with-registers-small
---
### Example
```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
```