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(
|