Visual-Recall / docker-entrypoint.sh
thesparshsaxena's picture
Corrupt images and delete function fix (#12)
f7d9bc5
Raw
History Blame Contribute Delete
1.28 kB
#!/usr/bin/env bash
set -euo pipefail
DATA_DIR="${DATA_DIR:-/data}"
APP_DIR="${APP_DIR:-/app}"
MODEL_DIR="$DATA_DIR/models"
DB_DIR="$DATA_DIR/dbs"
IMAGE_DIR="$DATA_DIR/encrypted_images"
BUNDLED_MODEL_DIR="$APP_DIR/data/models"
if ! command -v tesseract >/dev/null 2>&1; then
echo "tesseract binary is missing; installing tesseract-ocr and English language data." >&2
apt-get update
apt-get install -y --no-install-recommends tesseract-ocr tesseract-ocr-eng
rm -rf /var/lib/apt/lists/*
fi
mkdir -p "$DATA_DIR" "$MODEL_DIR" "$DB_DIR" "$IMAGE_DIR"
if [ -d "$BUNDLED_MODEL_DIR" ] && [ -z "$(find "$MODEL_DIR" -mindepth 1 -maxdepth 1 -print -quit)" ]; then
cp -a "$BUNDLED_MODEL_DIR"/. "$MODEL_DIR"/
fi
if [ -d "$APP_DIR/encrypted_images" ] && [ ! -L "$APP_DIR/encrypted_images" ]; then
find "$APP_DIR/encrypted_images" -mindepth 1 -maxdepth 1 -exec cp -a {} "$IMAGE_DIR"/ \;
rm -rf "$APP_DIR/encrypted_images"
elif [ -e "$APP_DIR/encrypted_images" ] && [ ! -L "$APP_DIR/encrypted_images" ]; then
rm -f "$APP_DIR/encrypted_images"
fi
ln -sfn "$IMAGE_DIR" "$APP_DIR/encrypted_images"
exec streamlit run Homepage.py \
--server.port=8501 \
--server.address=0.0.0.0 \
--server.enableCORS=false \
--server.enableXsrfProtection=false