Commit
•
1f9b3b5
1
Parent(s):
f7d1fb6
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
license: apache-2.0
|
| 3 |
library_name: transformers
|
| 4 |
pipeline_tag: image-classification
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
This is a simple AI image detection model utilizing visual transformers trained on the CIFake dataset.
|
|
@@ -51,4 +53,4 @@ image_paths = [
|
|
| 51 |
for image_path in image_paths:
|
| 52 |
predicted_class = predict(image_path, model, preprocess)
|
| 53 |
print(f'Predicted class: {predicted_class}', image_path)
|
| 54 |
-
```
|
|
|
|
| 2 |
license: apache-2.0
|
| 3 |
library_name: transformers
|
| 4 |
pipeline_tag: image-classification
|
| 5 |
+
metrics:
|
| 6 |
+
- accuracy: 98%
|
| 7 |
---
|
| 8 |
|
| 9 |
This is a simple AI image detection model utilizing visual transformers trained on the CIFake dataset.
|
|
|
|
| 53 |
for image_path in image_paths:
|
| 54 |
predicted_class = predict(image_path, model, preprocess)
|
| 55 |
print(f'Predicted class: {predicted_class}', image_path)
|
| 56 |
+
```
|