Use hash to keep TensorFlow model up to date

scripts/download-assets.sh

@@ -1,15 +0,0 @@
-#!/usr/bin/env bash
-
-FILENAME="/tmp/photoprism/nasnet.zip"
-
-if [[ ! -e assets/tensorflow/nasnet/saved_model.pb ]]; then
-  if [[ ! -e ${FILENAME} ]]; then
-      mkdir -p /tmp/photoprism
-      wget "https://dl.photoprism.org/tensorflow/nasnet.zip" -O ${FILENAME}
-  fi
-
-  mkdir -p assets/tensorflow
-  unzip ${FILENAME} -d assets/tensorflow
-else
-  echo "TensorFlow model already downloaded."
-fi

scripts/download-nasnet.sh

@@ -0,0 +1,45 @@
+#!/usr/bin/env bash
+
+TODAY=`date -u +%Y%m%d`
+
+MODEL_NAME="NASNet Mobile"
+MODEL_URL="https://dl.photoprism.org/tensorflow/nasnet.zip?$TODAY"
+MODEL_PATH="assets/tensorflow/nasnet"
+MODEL_ZIP="/tmp/photoprism/nasnet.zip"
+MODEL_HASH="6a9450f89afa56b4539c0d7188f108f083c10fc9  $MODEL_ZIP"
+MODEL_VERSION="$MODEL_PATH/version.txt"
+MODEL_BACKUP="assets/backups/nasnet-$TODAY"
+
+echo "Installing $MODEL_NAME for TensorFlow..."
+
+# Check for update
+if [[ -f ${MODEL_ZIP} ]] && [[ `sha1sum ${MODEL_ZIP}` == ${MODEL_HASH} ]]; then
+  echo "Already up to date."
+  exit
+fi
+
+# Create directories
+mkdir -p /tmp/photoprism
+mkdir -p assets/tensorflow
+mkdir -p assets/backups
+
+# Download model
+echo "Downloading latest model from $MODEL_URL..."
+wget ${MODEL_URL} -O ${MODEL_ZIP}
+
+TMP_HASH=`sha1sum ${MODEL_ZIP}`
+
+echo ${TMP_HASH}
+
+# Create backup
+if [[ -e ${MODEL_PATH} ]]; then
+  echo "Creating backup of existing directory: $MODEL_BACKUP"
+  rm -rf ${MODEL_BACKUP}
+  mv ${MODEL_PATH} ${MODEL_BACKUP}
+fi
+
+# Unzip model
+unzip ${MODEL_ZIP} -d assets/tensorflow
+echo "$MODEL_NAME $TODAY $MODEL_HASH" > ${MODEL_VERSION}
+
+echo "Latest $MODEL_NAME installed."