| --- |
| 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 |
| ``` |
|
|