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