File size: 1,074 Bytes
7c482d2 2cd9af9 7c482d2 2cd9af9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
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()
``` |