How to use sharktide/ohca-predictor-v1 with Keras:
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://sharktide/ohca-predictor-v1")
How to use sharktide/ohca-predictor-v1 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import TFAutoModel model = TFAutoModel.from_pretrained("sharktide/ohca-predictor-v1", trust_remote_code=True, device_map="auto")
The community tab is the place to discuss and collaborate with the HF community!
