Spaces:

B4869
/

text_embedding_model

Sleeping

text_embedding_model / download_model.py

Update download_model.py

67ed46c verified 11 months ago

324 Bytes

#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!")

	#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!")