scripts/publish_assets_from_local.sh

#!/usr/bin/env bash
set -euo pipefail

# Usage:
#   bash scripts/publish_assets_from_local.sh SalmaHassan/HyperClinical-assets ../avra_public-master/HyperClinical_Challenge

ASSETS_REPO_ID="${1:-SalmaHassan/HyperClinical-assets}"
SOURCE_ROOT="${2:-../avra_public-master/HyperClinical_Challenge}"

if [[ ! -f "${SOURCE_ROOT}/checkpoints/neurofusion/best_model.pt" ]]; then
  echo "Missing source checkpoint at ${SOURCE_ROOT}/checkpoints/neurofusion/best_model.pt"
  exit 1
fi

TMP_DIR="$(mktemp -d)"
trap 'rm -rf "${TMP_DIR}"' EXIT

mkdir -p "${TMP_DIR}/checkpoints/neurofusion"
mkdir -p "${TMP_DIR}/src/inference_core/weights"
mkdir -p "${TMP_DIR}/src/assets"

cp "${SOURCE_ROOT}/checkpoints/neurofusion/best_model.pt" "${TMP_DIR}/checkpoints/neurofusion/best_model.pt"
cp "${SOURCE_ROOT}/checkpoints/neurofusion/preprocessing_stats.json" "${TMP_DIR}/checkpoints/neurofusion/preprocessing_stats.json"
cp -a "${SOURCE_ROOT}/checkpoints/avra/weights/." "${TMP_DIR}/src/inference_core/weights/"
if [[ -f "${SOURCE_ROOT}/Hyperclinical_Pipeline.jpg" ]]; then
  cp "${SOURCE_ROOT}/Hyperclinical_Pipeline.jpg" "${TMP_DIR}/src/assets/Hyperclinical_Pipeline.jpg"
fi

# Requires: huggingface-cli login
# Create repo once if needed:
#   huggingface-cli repo create "${ASSETS_REPO_ID}" --type model

huggingface-cli upload-large-folder "${ASSETS_REPO_ID}" "${TMP_DIR}" --repo-type model

echo "Uploaded assets to ${ASSETS_REPO_ID}"