Update README.md

README.md

@@ -1,9 +1,9 @@
 ---
 language: en
 tags:
-- image-classification
-- ai-detection
-- vit
+  - image-classification
+  - ai-detection
+  - vit
 license: mit
 ---
 
@@ -39,4 +39,44 @@ def detect_image(image_path):
     confidence = predictions[0][prediction_id].item() * 100
     
     result = "AI Generated" if prediction_id == 1 else "Real Image"
-    return result, confidence
+    return result, confidence
+
+# Example usage
+# result, confidence = detect_image("path/to/image.jpg")
+# print(f"Result: {result} (Confidence: {confidence:.2f}%)")
+Classes
+The model classifies images into two categories:
+
+Real Image (0)
+AI Generated (1)
+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
+CopyInsert
+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