mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-19 21:37:31 +08:00
10 lines
335 B
Python
10 lines
335 B
Python
from decouple import config
|
|
from sentence_transformers import SentenceTransformer
|
|
|
|
EMBEDDING_MODEL = config("EMBEDDING_MODEL", cast=str, default="paraphrase-multilingual-mpnet-base-v2")
|
|
|
|
# Initialize embedding model
|
|
model = SentenceTransformer(EMBEDDING_MODEL)
|
|
|
|
model.save("./transformer_model/paraphrase-multilingual-mpnet-base-v2")
|