Update README.md

README.md

@@ -10,7 +10,7 @@ license: mit
 # AI Image Detector
 
 ## Model Description
-This model is designed to detect whether an image is real or AI-generated. It uses Vision Transformer (VIT) architecture to provide accurate classification.
+This model is designed to detect whether an image is real or AI-generated. It uses Vision Transformer (ViT) architecture to provide accurate classification.
 
 ## Model Usage
 ```python
@@ -43,4 +43,41 @@ def detect_image(image_path):
 
 # Example usage
 # result, confidence = detect_image("path/to/image.jpg")
-# print(f"Result: {result} (Confidence: {confidence:.2f}%)")
+# print(f"Result: {result} (Confidence: {confidence:.2f}%)")
+Classes
+The model classifies images into two categories:
+
+Real Image (0): The image is real and not AI-generated.
+AI Generated (1): The image is generated by AI.
+Technical Details
+Model Architecture: Vision Transformer (ViT)
+Input: Images (RGB)
+Output: Binary classification with confidence score
+Max Image Size: 224x224 (automatically resized)
+Requirements
+transformers>=4.30.0
+torch>=2.0.0
+Pillow>=9.0.0
+Limitations
+Best performance with clear, high-quality images.
+May have reduced accuracy with heavily edited photos.
+Designed for general image detection.
+Web Integration Example
+javascript
+async function detectImage(imageFile) {
+  const formData = new FormData();
+  formData.append('image', imageFile);
+
+  const response = await fetch('YOUR_API_ENDPOINT', {
+    method: 'POST',
+    body: formData
+  });
+
+  return await response.json();
+}
+Developer
+Created by: yaya36095
+License: MIT
+Repository: https://huggingface.co/yaya36095/ai-image-detector
+markdown
+