metadata
license: apache-2.0
tags:
- image-classification
- keras
- tensorflow
- alertdrive
AlertDrive AI Model
Binary classification model for driver alert detection.
Model Details
- Input: 224x224x3 RGB images
- Output: Binary classification (Alert/No Alert)
- Framework: TensorFlow/Keras
Usage
from huggingface_hub import hf_hub_download
import tensorflow as tf
# Download model
model_path = hf_hub_download(repo_id="YOUR_USERNAME/alertdrive-model", filename="alertdrive_ai.keras")
model = tf.keras.models.load_model(model_path)
# Make prediction
prediction = model.predict(image_array)