lamhieu commited on
Commit
24dd113
Β·
1 Parent(s): 4252268

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="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
114
- onnx_file="onnx/model_quint8_avx2.onnx",
115
  ),
116
  TextModelType.PARAPHRASE_MULTILINGUAL_MPNET_BASE_V2: ModelInfo(
117
- model_id="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
118
- onnx_file="onnx/model_quint8_avx2.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",