text_embedding_model / download_model.py
B4869's picture
Update download_model.py
67ed46c verified
#login huggingface
from huggingface_hub import login
import os
HF_TOKEN = os.getenv("HF_TOKEN")
login(HF_TOKEN)
#download model
from sentence_transformers import SentenceTransformer
print("Downloading model...")
model = SentenceTransformer("google/embeddinggemma-300m")
print("Model downloaded and cached successfully!")