Keras
model-api / README.md
nicholasKluge's picture
Update README.md
2cd9af9 verified
---
license: apache-2.0
library_name: keras
---
# Model API (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
import json
import tensorflow as tf
from huggingface_hub import hf_hub_download
# Download the model (this will be the target of our attack)
hf_hub_download(repo_id="AiresPucrs/model-api",
                filename="model_api.h5",
                local_dir="./",
                repo_type="model"
)
# Download the tokenizer file
hf_hub_download(repo_id="AiresPucrs/model-api",
                filename="tokenizer_model_api.json",
                local_dir="./",
                repo_type="model"
)
model_api = tf.keras.models.load_model('./model_api.h5')
with open('./tokenizer_model_api.json') as fp:
data = json.load(fp)
tokenizer = tf.keras.preprocessing.text.tokenizer_from_json(data)
fp.close()
```