| --- |
| 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() |
| ``` |