metadata

title: AI Image Detector
emoji: 🤖
colorFrom: purple
colorTo: indigo
sdk: gradio
sdk_version: 5.34.2
app_file: app.py
pinned: false
license: mit

🤖 AI Image Detector

Detect whether an image is AI-generated or real using state-of-the-art machine learning models.

Overview

This Gradio app uses a specialized model to classify images as either AI-generated or real. The model has been specifically trained to detect images generated by various AI systems including:

DALL-E
Midjourney
Stable Diffusion (SDXL)
And other diffusion models

How to Use

Upload an Image: Click on the upload area or drag and drop an image file (JPG, PNG, etc.)
Get Results: The model will analyze your image and return probability scores
Interpret Results: Higher probability for "AI-generated" suggests the image was created by AI

Model Information

Task: Image Classification (Binary: AI-generated vs Real)
Framework: Transformers + PyTorch
Interface: Gradio

Limitations

⚠️ Important Notes:

The model may not be 100% accurate on all images
Performance may vary depending on the specific AI model used to generate the image
Very high-quality AI images or heavily post-processed real images might be misclassified
The model is primarily trained on SDXL-style generated images

Technical Details

The app uses direct model inference to provide robust classification results. The model outputs probabilities for each class, giving you confidence scores for the prediction.

Development

This space is built with:

Gradio: For the web interface
Transformers: For model loading and inference
PyTorch: As the backend framework

This is an educational tool for demonstrating AI image detection capabilities. Always use critical thinking when evaluating image authenticity.