mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-28 01:43:18 +08:00
7 lines
261 B
Python
7 lines
261 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) |