Replace Gemma embedding model with BAAI/bge-small-en-v1.5 and cache at build time

Dockerfile

@@ -21,6 +21,7 @@ RUN mkdir -p /app/nltk_data && \
 RUN python -c "from huggingface_hub import snapshot_download; \
     snapshot_download(repo_id='nomic-ai/nomic-embed-text-v1.5'); \
     snapshot_download(repo_id='C10X/Qwen3-Embedding-TurboX.v2'); \
+    snapshot_download(repo_id='BAAI/bge-small-en-v1.5'); \
     snapshot_download(repo_id='Qwen/Qwen2.5-0.5B-Instruct')"
 
 

backend/constants.py

@@ -95,6 +95,6 @@ Passage:"""
 
 MODEL_MAP = {
     "nomic-embed-text": "nomic-ai/nomic-embed-text-v1.5",
-    "EmbeddingGemma": "google/embeddinggemma-300m",
+    "bge-small-en": "BAAI/bge-small-en-v1.5",
     "qwen3-embedding:0.6b": "C10X/Qwen3-Embedding-TurboX.v2",
 }

frontend/index.html

@@ -168,7 +168,7 @@
               <label for="embedding-model">Embedding Model</label>
               <select id="embedding-model" class="select-input">
                 <option value="nomic-embed-text" selected>Nomic Embed Text</option>
-                <option value="EmbeddingGemma">Embedding Gemma</option>
+                <option value="bge-small-en">BGE Small English</option>
                 <option value="qwen3-embedding:0.6b">Qwen3 Embedding</option>
               </select>
             </div>
@@ -375,7 +375,7 @@
                     </select>
                     <select id="arena-model-a" class="select-input">
                       <option value="nomic-embed-text">Nomic Embed</option>
-                      <option value="EmbeddingGemma">Gemma Embed</option>
+                      <option value="bge-small-en">BGE Small</option>
                       <option value="qwen3-embedding:0.6b">Qwen3 Embed</option>
                     </select>
                     <select id="arena-strategy-a" class="select-input">
@@ -404,7 +404,7 @@
                     </select>
                     <select id="arena-model-b" class="select-input">
                       <option value="nomic-embed-text">Nomic Embed</option>
-                      <option value="EmbeddingGemma">Gemma Embed</option>
+                      <option value="bge-small-en">BGE Small</option>
                       <option value="qwen3-embedding:0.6b">Qwen3 Embed</option>
                     </select>
                     <select id="arena-strategy-b" class="select-input">