Spaces:
Running
Running
chore: update model ids
Browse files
lightweight_embeddings/__init__.py
CHANGED
@@ -56,7 +56,6 @@ EMBEDDINGS_API_URL = "http://localhost:7860/v1/embeddings"
|
|
56 |
|
57 |
# Markdown description for the main interface
|
58 |
APP_DESCRIPTION = f"""
|
59 |
-
<br />
|
60 |
## π **Lightweight Embeddings API**
|
61 |
|
62 |
The **Lightweight Embeddings API** is a fast, free, and multilingual service designed for generating embeddings and reranking with support for both **text** and **image** inputs. Get started below by exploring our interactive playground or using the cURL examples provided.
|
|
|
56 |
|
57 |
# Markdown description for the main interface
|
58 |
APP_DESCRIPTION = f"""
|
|
|
59 |
## π **Lightweight Embeddings API**
|
60 |
|
61 |
The **Lightweight Embeddings API** is a fast, free, and multilingual service designed for generating embeddings and reranking with support for both **text** and **image** inputs. Get started below by exploring our interactive playground or using the cURL examples provided.
|
lightweight_embeddings/service.py
CHANGED
@@ -110,12 +110,12 @@ class ModelConfig:
|
|
110 |
onnx_file="onnx/model_quantized.onnx",
|
111 |
),
|
112 |
TextModelType.PARAPHRASE_MULTILINGUAL_MINILM_L12_V2: ModelInfo(
|
113 |
-
model_id="
|
114 |
-
onnx_file="onnx/
|
115 |
),
|
116 |
TextModelType.PARAPHRASE_MULTILINGUAL_MPNET_BASE_V2: ModelInfo(
|
117 |
-
model_id="
|
118 |
-
onnx_file="onnx/
|
119 |
),
|
120 |
TextModelType.BGE_M3: ModelInfo(
|
121 |
model_id="BAAI/bge-m3",
|
|
|
110 |
onnx_file="onnx/model_quantized.onnx",
|
111 |
),
|
112 |
TextModelType.PARAPHRASE_MULTILINGUAL_MINILM_L12_V2: ModelInfo(
|
113 |
+
model_id="Xenova/paraphrase-multilingual-MiniLM-L12-v2",
|
114 |
+
onnx_file="onnx/model_quantized.onnx",
|
115 |
),
|
116 |
TextModelType.PARAPHRASE_MULTILINGUAL_MPNET_BASE_V2: ModelInfo(
|
117 |
+
model_id="Xenova/paraphrase-multilingual-mpnet-base-v2",
|
118 |
+
onnx_file="onnx/model_quantized.onnx",
|
119 |
),
|
120 |
TextModelType.BGE_M3: ModelInfo(
|
121 |
model_id="BAAI/bge-m3",
|