Spaces:
Running
Running
Patryk Ptasiński
commited on
Commit
·
1ea76a1
1
Parent(s):
eca0fa7
Add Qwen3 and Stella embedding models to predefined list
Browse files- Add Qwen3-Embedding-0.6B, 4B, and 8B models
- Add Stella models: dunzhang/stella_en_400M_v5, stella_en_1.5B_v5, infgrad/stella-base-en-v2
- Update model count to 22+ in description
app.py
CHANGED
|
@@ -24,6 +24,12 @@ MODELS = {
|
|
| 24 |
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2": {"trust_remote_code": False},
|
| 25 |
"ibm-granite/granite-embedding-30m-english": {"trust_remote_code": False},
|
| 26 |
"ibm-granite/granite-embedding-278m-multilingual": {"trust_remote_code": False},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
}
|
| 28 |
|
| 29 |
# Model cache
|
|
@@ -124,7 +130,7 @@ with gr.Blocks(title="Multi-Model Text Embeddings", css="""
|
|
| 124 |
}
|
| 125 |
""") as app:
|
| 126 |
gr.Markdown("# Multi-Model Text Embeddings")
|
| 127 |
-
gr.Markdown("Generate embeddings for your text using
|
| 128 |
|
| 129 |
# Model selector dropdown (allows custom input)
|
| 130 |
model_dropdown = gr.Dropdown(
|
|
|
|
| 24 |
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2": {"trust_remote_code": False},
|
| 25 |
"ibm-granite/granite-embedding-30m-english": {"trust_remote_code": False},
|
| 26 |
"ibm-granite/granite-embedding-278m-multilingual": {"trust_remote_code": False},
|
| 27 |
+
"Qwen/Qwen3-Embedding-0.6B": {"trust_remote_code": False},
|
| 28 |
+
"Qwen/Qwen3-Embedding-4B": {"trust_remote_code": False},
|
| 29 |
+
"Qwen/Qwen3-Embedding-8B": {"trust_remote_code": False},
|
| 30 |
+
"dunzhang/stella_en_400M_v5": {"trust_remote_code": False},
|
| 31 |
+
"dunzhang/stella_en_1.5B_v5": {"trust_remote_code": False},
|
| 32 |
+
"infgrad/stella-base-en-v2": {"trust_remote_code": False},
|
| 33 |
}
|
| 34 |
|
| 35 |
# Model cache
|
|
|
|
| 130 |
}
|
| 131 |
""") as app:
|
| 132 |
gr.Markdown("# Multi-Model Text Embeddings")
|
| 133 |
+
gr.Markdown("Generate embeddings for your text using 22+ state-of-the-art embedding models from Nomic, BGE, Snowflake, IBM Granite, Qwen3, Stella, and more.")
|
| 134 |
|
| 135 |
# Model selector dropdown (allows custom input)
|
| 136 |
model_dropdown = gr.Dropdown(
|