weights_dl.sh

#!/usr/bin/env bash
set -e

# Script to download weight files from Google Drive
# Folder ID: 1Mj7TE2IH8EtHBIswu8TJ9tjBfzYG2Pv2

FOLDER_ID="1Mj7TE2IH8EtHBIswu8TJ9tjBfzYG2Pv2"
OUTPUT_DIR="${1:-.}"

# Check if gdown is installed
if ! command -v gdown &> /dev/null; then
    echo "gdown is not installed. Installing..."
    pip install gdown
fi

# Create output directory
mkdir -p "${OUTPUT_DIR}"

echo "Downloading weight files from Google Drive..."
echo "Folder ID: ${FOLDER_ID}"
echo "Output directory: ${OUTPUT_DIR}"

# Download Google Drive folder
# Try multiple methods as syntax may differ depending on gdown version
if gdown --folder "https://drive.google.com/drive/folders/${FOLDER_ID}" -O "${OUTPUT_DIR}" 2>/dev/null; then
    echo "Download completed!"
elif gdown --folder "${FOLDER_ID}" -O "${OUTPUT_DIR}" 2>/dev/null; then
    echo "Download completed!"
else
    echo "Error: Failed to download folder."
    echo "Please check gdown version: gdown --version"
    echo "To upgrade to latest version: pip install --upgrade gdown"
    exit 1
fi