# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras
model = keras.saving.load_model("hf://Dulya8/vit-model")
SquintVue ViT Inference Endpoint
This repository contains the deployed Vision Transformer inference model for SquintVue.
Model
- File:
strabismus_vit.keras - Framework: TensorFlow / Keras
- Backbone: ViT Base Patch16 224
- Classes:
- ESOTROPIA
- EXOTROPIA
- HYPERTROPIA
- HYPOTROPIA
- UBIPR_Healthy_Eyes
Input
JSON request body:
{ "inputs": { "image_base64": "BASE64_ENCODED_IMAGE" } }
Output
{ "predicted_class": "UBIPR_Healthy_Eyes", "confidence": 0.9631, "confidence_percentage": 96.31, "class_probabilities": [ { "label": "ESOTROPIA", "probability": 0.0086, "percentage": 0.86 } ], "model_input_size": [224, 224], "xai_ready": false }
- Downloads last month
- 4
# Gated model: Login with a HF token with gated access permission hf auth login