Update README.md
Browse files
README.md
CHANGED
@@ -21,6 +21,20 @@ language:
|
|
21 |
|
22 |
> deepfake-detector-model is a vision-language model fine-tuned from `google/vit-base-patch16-224-in21k` for binary image classification. It is trained to detect whether an image is fake or real using the *OpenDeepfake-Preview* dataset. The model uses the `ViTForImageClassification` architecture.
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
---
|
25 |
|
26 |
## Label Space: 2 Classes
|
|
|
21 |
|
22 |
> deepfake-detector-model is a vision-language model fine-tuned from `google/vit-base-patch16-224-in21k` for binary image classification. It is trained to detect whether an image is fake or real using the *OpenDeepfake-Preview* dataset. The model uses the `ViTForImageClassification` architecture.
|
23 |
|
24 |
+
```py
|
25 |
+
Classification Report:
|
26 |
+
precision recall f1-score support
|
27 |
+
|
28 |
+
fake 0.9817 0.9748 0.9782 10000
|
29 |
+
real 0.9750 0.9818 0.9784 9999
|
30 |
+
|
31 |
+
accuracy 0.9783 19999
|
32 |
+
macro avg 0.9783 0.9783 0.9783 19999
|
33 |
+
weighted avg 0.9783 0.9783 0.9783 19999
|
34 |
+
```
|
35 |
+
|
36 |
+

|
37 |
+
|
38 |
---
|
39 |
|
40 |
## Label Space: 2 Classes
|