Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,11 @@ language:
|
|
| 7 |
metrics:
|
| 8 |
- accuracy
|
| 9 |
pipeline_tag: image-classification
|
| 10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
metrics:
|
| 8 |
- accuracy
|
| 9 |
pipeline_tag: image-classification
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
This is a model that uses transform learning to train an exising model on image dataset called "aiornot".
|
| 13 |
+
The model used as base it VGG16.
|
| 14 |
+
The task of the trained model is to predict if an image is generated by AI or not.
|
| 15 |
+
|
| 16 |
+
The file structure is as follows:
|
| 17 |
+
* One notebook file containing all the code.
|