Spaces:
Runtime error
Runtime error
Create baai_bge embeeding
Browse files
api/embedding_models/baai_bge_m3.py
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from llama_index.embeddings.huggingface import HuggingFaceEmbedding
|
| 3 |
+
from api.embedding_models.embedding_config import EmbeddingConfig
|
| 4 |
+
from llama_index.core import Settings
|
| 5 |
+
|
| 6 |
+
root_path = os.environ['ROOT_PATH']
|
| 7 |
+
MODEL_DIRECTORY = os.path.join(root_path,"model")
|
| 8 |
+
|
| 9 |
+
def EmbeddingModel():
|
| 10 |
+
current_file_path = os.path.abspath(__file__)
|
| 11 |
+
embedding_model_dir = os.path.dirname(current_file_path)
|
| 12 |
+
api_dir = os.path.dirname(embedding_model_dir)
|
| 13 |
+
model_dir = os.path.join(api_dir,MODEL_DIRECTORY)
|
| 14 |
+
Settings.embed_model = HuggingFaceEmbedding(model_name = "BAAI/bge-m3",
|
| 15 |
+
cache_folder = model_dir)
|
| 16 |
+
return EmbeddingConfig()
|
| 17 |
+
|