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

Files changed (1) hide show
  1. app.py +7 -1
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 16+ state-of-the-art embedding models from Nomic, BGE, Snowflake, IBM Granite, and more.")
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(