Instructions to use peterhuang24/AI_Generated_Image_Classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use peterhuang24/AI_Generated_Image_Classifier with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://peterhuang24/AI_Generated_Image_Classifier") - Notebooks
- Google Colab
- Kaggle
Upload config.json
Browse files- config.json +2 -5
config.json
CHANGED
|
@@ -1,8 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "
|
| 3 |
"input_shape": [224, 224, 3],
|
| 4 |
-
"num_classes": 1
|
| 5 |
-
"architecture": "ResNet152",
|
| 6 |
-
"description": "Fine-tuned ResNet152 for binary classification"
|
| 7 |
-
|
| 8 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"model_type": "tensorflow",
|
| 3 |
"input_shape": [224, 224, 3],
|
| 4 |
+
"num_classes": 1
|
|
|
|
|
|
|
|
|
|
| 5 |
}
|