chore: update something

README.md

@@ -21,8 +21,8 @@ header: mini
 - **Reranking Support**: Includes powerful reranking capabilities for both text and image inputs.
 - **Optimized for Speed**: Built with lightweight transformer models and efficient backends for rapid inference, even on low-resource systems.
 - **Flexible Model Support**: Use a range of transformer models tailored to diverse use cases:
-  - Text models: `multilingual-e5-small`, `paraphrase-multilingual-MiniLM-L12-v2`, `bge-m3`
-  - Image model: `google/siglip-base-patch16-256-multilingual`
+  - Text models: `multilingual-e5-small`, `multilingual-e5-base`, `multilingual-e5-large`, `snowflake-arctic-embed-l-v2.0`, `paraphrase-multilingual-MiniLM-L12-v2`, `paraphrase-multilingual-mpnet-base-v2`, `bge-m3`.
+  - Image model: `siglip-base-patch16-256-multilingual`
 - **Production-Ready**: Easily deploy anywhere with Docker for hassle-free setup.
 - **Interactive Playground**: Test embeddings and reranking directly via a **Gradio-powered interface** alongside detailed REST API documentation.
 

lightweight_embeddings/__init__.py

@@ -7,7 +7,11 @@ This application provides text and image embeddings using multiple text models a
 
 Supported text model IDs:
 - "multilingual-e5-small"
+- "multilingual-e5-base"
+- "multilingual-e5-large"
+- "snowflake-arctic-embed-l-v2.0"
 - "paraphrase-multilingual-MiniLM-L12-v2"
+- "paraphrase-multilingual-mpnet-base-v2"
 - "bge-m3"
 
 Supported image model ID:
@@ -130,10 +134,10 @@ def create_main_interface():
     """
     # Available model options for the dropdown
     model_options = [
+        "snowflake-arctic-embed-l-v2.0",
         "multilingual-e5-small",
         "multilingual-e5-base",
         "multilingual-e5-large",
-        "snowflake-arctic-embed-l-v2.0",
         "paraphrase-multilingual-MiniLM-L12-v2",
         "paraphrase-multilingual-mpnet-base-v2",
         "bge-m3",
@@ -184,7 +188,7 @@ def create_main_interface():
                     -H 'accept: application/json' \\
                     -H 'Content-Type: application/json' \\
                     -d '{
-                    "model": "multilingual-e5-small",
+                    "model": "snowflake-arctic-embed-l-v2.0",
                     "input": "That is a happy person"
                   }'
                   ```
@@ -196,7 +200,7 @@ def create_main_interface():
                     -H 'accept: application/json' \\
                     -H 'Content-Type: application/json' \\
                     -d '{
-                    "model": "multilingual-e5-small",
+                    "model": "snowflake-arctic-embed-l-v2.0",
                     "queries": "That is a happy person",
                     "candidates": [
                       "That is a happy dog",

lightweight_embeddings/router.py

@@ -5,12 +5,15 @@ Exposes the EmbeddingsService methods via a RESTful API.
 
 Supported Text Model IDs:
 - "multilingual-e5-small"
+- "multilingual-e5-base"
+- "multilingual-e5-large"
+- "snowflake-arctic-embed-l-v2.0"
 - "paraphrase-multilingual-MiniLM-L12-v2"
+- "paraphrase-multilingual-mpnet-base-v2"
 - "bge-m3"
 
 Supported Image Model IDs:
 - "siglip-base-patch16-256-multilingual"
-(Extend as needed)
 """
 
 from __future__ import annotations

lightweight_embeddings/service.py

@@ -12,7 +12,11 @@ Features:
 
 Supported Text Model IDs:
 - "multilingual-e5-small"
+- "multilingual-e5-base"
+- "multilingual-e5-large"
+- "snowflake-arctic-embed-l-v2.0"
 - "paraphrase-multilingual-MiniLM-L12-v2"
+- "paraphrase-multilingual-mpnet-base-v2"
 - "bge-m3"
 
 Supported Image Model IDs: