#!/bin/bash

echo "Note: available models are carson_Jan2021, carmen_Jan2021"

echo "Downloading weights from the Harvard Dataverse: https://doi.org/10.7910/DVN/XB6PEZ"

# Define the URLs of the files to download

url1="https://dataverse.harvard.edu/api/access/datafile/10358815"

url2="https://dataverse.harvard.edu/api/access/datafile/10358816"

# Define the directory to save the files

MODEL_DIR="./pretrained_models"

# Define the filenames

filename1="carmen_Jan2021.h5"

filename2="carson_Jan2021.h5"

# Create the directory if it doesn't exist

mkdir -p "$MODEL_DIR"

# Function to download a file

download_file() {

  local url=$1

  local filename=$2

  curl -L -o "$MODEL_DIR/$filename" "$url"

  # Check if the download was successful

  if [ $? -eq 0 ]; then

      echo "Download of $filename completed successfully. Saved to $MODEL_DIR/$filename"

  else

      echo "Download of $filename failed."

      exit 1

  fi

}

# Download the files

download_file "$url1" "$filename1"

download_file "$url2" "$filename2"