Spaces:

Vasanth6
/

rag-visualizer

Sleeping

Vasanth6 commited on Jun 25

Commit

6bac071

1 Parent(s): 1142544

Update schemas.py EmbeddingModel enum to replace EmbeddingGemma with bge-small-en

Files changed (1) hide show

backend/models/schemas.py CHANGED Viewed

@@ -16,7 +16,7 @@ class Strategy(str, Enum):
 class EmbeddingModel(str, Enum):
     NOMIC_EMBED_TEXT = "nomic-embed-text"
-    EMBEDDING_GEMMA = "EmbeddingGemma"
     QWEN_EMBEDDING = "qwen3-embedding:0.6b"

 class EmbeddingModel(str, Enum):
     NOMIC_EMBED_TEXT = "nomic-embed-text"
+    BGE_SMALL_EN = "bge-small-en"
     QWEN_EMBEDDING = "qwen3-embedding:0.6b"