File size: 842 Bytes
c21a2c7 9605716 c21a2c7 9605716 2ab8001 9605716 c21a2c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from transformers import pipeline
import gradio as gr
classifier = pipeline("image-classification", model="SADRACODING/SDXL-Deepfake-Detector")
def predict(image):
if image is None:
return None
results = classifier(image)
return results[0]["label"]
gr.Interface(
fn=predict,
inputs=gr.Image(type="pil", label="Upload an Image"),
outputs=gr.Label(label="Prediction"),
title="🔍 SDXL Deepfake Detector",
description="Upload a face image to detect if it's AI-generated (deepfake) or real.",
examples=[
["example_fake1.jpg"],
["example_fake2.jpg"],
["example_fake3.jpg"],
["example_fake4.jpg"],
["example_real1.jpg"],
["example_real2.jpg"],
["example_real3.jpg"],
["example_real4.jpg"],
["example_real5.jpg"]
]
).launch() |