#!/bin/bash
# Script to download the model weights from Hugging Face.
# This ensures that users don't have to clone massive model files unless needed.

echo "Downloading model weights from Hugging Face (kilanisainikhil/AerialEye)..."

# Download PT model
echo "Downloading aerialEye.pt..."
wget -c "https://huggingface.co/kilanisainikhil/AerialEye/resolve/main/aerialEye.pt" -O aerialEye.pt

# Download ONNX model
echo "Downloading aerialEye.onnx..."
wget -c "https://huggingface.co/kilanisainikhil/AerialEye/resolve/main/aerialEye.onnx" -O aerialEye.onnx

# Download Best PT model
echo "Downloading best.pt..."
wget -c "https://huggingface.co/kilanisainikhil/AerialEye/resolve/main/best.pt" -O best.pt

echo "Download complete!"