#!/bin/bash

# Ensure huggingface-cli is installed
if ! command -v huggingface-cli &> /dev/null
then
    echo "huggingface-cli could not be found. Installing transformers."
    pip install transformers
fi

# Create directories if they do not exist
mkdir -p Mistral-7B-Instruct-v0.3 mt5-small

# Download Mistral-7B-Instruct-v0.3 model from Hugging Face
huggingface-cli download mistralai/Mistral-7B-Instruct-v0.3 --local-dir ./Mistral-7B-Instruct-v0.3

# Download mt5-small model from Hugging Face
huggingface-cli download google/mt5-small --local-dir ./mt5-small

echo "Download complete!"