Update README.md

README.md

@@ -15,14 +15,13 @@ This model is designed to detect whether an image is real or AI-generated. It us
 
 ## Model Usage
 ```python
-```python
 from transformers import ViTImageProcessor, ViTForImageClassification
 from PIL import Image
 import torch
 
 # تحميل النموذج والمعالج
-processor = ViTImageProcessor.from_pretrained("yaya36095/ai-image-detector")
-model = ViTForImageClassification.from_pretrained("yaya36095/ai-image-detector")
+processor = ViTImageProcessor.from_pretrained("C:/Users/SUPREME TECH/Desktop/SAM3/ai-image-detector")
+model = ViTForImageClassification.from_pretrained("C:/Users/SUPREME TECH/Desktop/SAM3/ai-image-detector")
 
 def detect_image(image_path):
     # فتح وتجهيز الصورة
@@ -57,12 +56,22 @@ def detect_image(image_path):
         ]
     }
 
-# مثال للاستخدام
-# result = detect_image("path/to/image.jpg")
-# print(f"النتيجة: {result['prediction']}")
-# print(f"درجة الثقة: {result['confidence']}")
-# print("التفاصيل:", *result['detailed_scores'], sep="\n- ")
-
+# كود للاختبار
+if __name__ == "__main__":
+    # يمكنك تغيير مسار الصورة هنا
+    image_path = "path/to/your/image.jpg"
+    
+    try:
+        result = detect_image(image_path)
+        print("\nنتائج تحليل الصورة:")
+        print(f"التصنيف: {result['prediction']}")
+        print(f"درجة الثقة: {result['confidence']}")
+        print("\nالتفاصيل:")
+        for score in result['detailed_scores']:
+            print(f"- {score}")
+            
+    except Exception as e:
+        print(f"حدث خطأ: {str(e)}")
 ```
 
 ## Classes