| 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() |