How to use from the
Use from the
Keras library
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"

import keras

model = keras.saving.load_model("hf://sharktide/FireTrustNet")

This model requires the following custom objects:

import tensorflow as tf
from tensorflow.keras.saving import register_keras_serializable
from tensorflow.keras import layers, models, backend as K
import numpy as np

@register_keras_serializable()
def firetrust_activation(x):
    return 0.5 + tf.sigmoid(x)  # output in range [0.5, 1.5]
Downloads last month
-
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Spaces using sharktide/FireTrustNet 2

Collection including sharktide/FireTrustNet