|
--- |
|
library_name: keras |
|
tags: |
|
- Image Classification |
|
language: |
|
- en |
|
metrics: |
|
- accuracy |
|
--- |
|
# Cats vs Dogs CNN (Teeny-Tiny Castle) |
|
|
|
This model is part of a tutorial tied to the [Teeny-Tiny Castle](https://github.com/Nkluge-correa/TeenyTinyCastle), an open-source repository containing educational tools for AI Ethics and Safety research. |
|
|
|
## How to Use |
|
|
|
```python |
|
!pip install huggingface_hub["tensorflow"] -q |
|
|
|
# We are loading our Keras model from the Hub via the `huggingface_hub.from_pretrained_keras` |
|
from huggingface_hub import from_pretrained_keras |
|
import tensorflow as tf |
|
|
|
model = from_pretrained_keras("AiresPucrs/cats-vs-dogs") |
|
model.trainable = False |
|
model.summary() |
|
``` |
|
|
