object-detection

# all default common env var values are defined in scripts/envs/base file
# in order to override a default common env var value and/or add a custom
# env var used only for this sample you can list it here
export APP_NAME="object-detection"
export APP_DISPLAY_NAME="Object Detection Application"
export APP_DESC="Identify and locate objects in an image using an AI-enabled object detection application. Pick from the DEtection TRansformer (DETR) model servers or bring your own DEtection TRansformer (DETR)."
export APP_SUMMARY="A DEtection TRansformer (DETR) model streamlit application. Upload an image to identify and locate objects in the image."
export APP_TAGS='["ai", "detr", "python"]'
export APP_RUN_COMMAND="streamlit run object_detection_client.py"

# https://github.com/redhat-ai-dev/developer-images/tree/main/model/detr-resnet-101%20
export INIT_CONTAINER="quay.io/redhat-ai-dev/detr-resnet-101:latest"
export INIT_CONTAINER_COMMAND="['cp', '-R', '/model/detr-resnet-101', '/shared/detr-resnet-101']"

# https://github.com/containers/ai-lab-recipes/tree/main/model_servers/object_detection_python
export MODEL_SERVICE_CONTAINER="quay.io/redhat-ai-dev/object_detection_python:latest"
export MODEL_SERVICE_DESC="A simple FastAPI application for use with the DEtection TRansformer (DETR) models"
export MODEL_SERVICE_SRC="https://github.com/containers/ai-lab-recipes/tree/main/model_servers/object_detection_python"
export MODEL_SERVICE_PORT=8000
export MODEL_PATH="/model/detr-resnet-101"

# https://huggingface.co/facebook/detr-resnet-101
export DETR_MODEL_NAME="facebook/detr-resnet-101"
export DETR_MODEL_CLASSIFICATION="Object Detection"
export DETR_MODEL_LICENSE="Apache-2.0"
export DETR_MODEL_SRC="https://huggingface.co/facebook/detr-resnet-101"

# model configurations
export SUPPORT_DETR=true

# for gitlab case, since gitlab does not have pipeline webhook pre-set to trigger the initial build
export APP_INTERFACE_CONTAINER="quay.io/redhat-ai-dev/object_detection:latest"

export TEMPLATE_SOURCE_URL="https://github.com/redhat-ai-dev/ai-lab-samples/tree/main/object-detection"

# application configuration
# Reference the above info as different names for templating purposes.
# Should probably be refactored.
export MODEL_NAME="${DETR_MODEL_NAME}"
export MODEL_SRC="${DETR_MODEL_SRC}"
export MODEL_SERVICE_SRC_OTHER="${MODEL_SERVICE_SRC}"