AiresPucrs/model-api

Model API (Teeny-Tiny Castle)

This model is part of a tutorial tied to the Teeny-Tiny Castle, an open-source repository containing educational tools for AI Ethics and Safety research.

How to Use

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