Instructions to use CapstoneML/Model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use CapstoneML/Model with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://CapstoneML/Model") - Notebooks
- Google Colab
- Kaggle
| import keras | |
| from keras.models import model_from_json | |
| print("keras versio:", keras.__version__) | |
| def load_model_from_files(json_path, weights_path): | |
| with open(json_path, "r") as json_file: | |
| loaded_model_json = json_file.read() | |
| model = model_from_json(loaded_model_json) | |
| model.load_weights(weights_path) | |
| model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) | |
| return model | |