# download_models.py
import os
from huggingface_hub import hf_hub_download
import torchvision.models as models

def download_all():
    print("Download ONNX models...")
    model_id = "onnx-community/embeddinggemma-300m-ONNX"
    # 토큰은 공개 모델이면 필요 없으나, 필요시 빌드 아규먼트로 처리해야 함
    hf_hub_download(model_id, subfolder="onnx", filename="model.onnx")
    hf_hub_download(model_id, subfolder="onnx", filename="model.onnx_data")
    
    print("Download Vision models...")
    models.efficientnet_b1(weights='IMAGENET1K_V1')
    print("Download Complete.")

if __name__ == "__main__":
    download_all()